
    &`i                        d dl Z d dlZd dlZd dlmZ d dlZd dlmZ d dlmZ ej	        dk    r ej
        d            nd dlmZ d dlmZ d Zd	 Zed
k    rm e j                    Ze                    ddd           e                                \  ZZej        r ej        d            eej        rdnd           dS dS )    N)FileLock)tune)AsyncHyperBandScheduler)      )mnist)TuneReportCheckpointCallbackc                    dd l }d}d}d}t          t          j                            d                    5  t          j                    \  \  }}\  }}d d d            n# 1 swxY w Y   |dz  |dz  }}|j        j        	                    |j        j
                            d          |j        j
                            | d	         d
          |j        j
                            d          |j        j
                            |d          g          }	|	                    d|j        j                            | d         | d                   dg           |	                    ||||d||ft%          g ddi          g           d S )Nr      
   r   z~/.data.lockg     o@)   r   )input_shapehiddenrelu)
activationg?softmaxsparse_categorical_crossentropylrmomentum)r   r   accuracy)loss	optimizermetricsmean_accuracy)checkpoint_onr   )
batch_sizeepochsverbosevalidation_data	callbacks)
tensorflowr   ospath
expanduserr   	load_datakerasmodels
SequentiallayersFlattenDenseDropoutcompile
optimizersSGDfitr	   )
configtfr   num_classesr   x_trainy_trainx_testy_testmodels
             v/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/ray/tune/examples/tune_mnist_keras.pytrain_mnistr:      s   JKF	"'$$^44	5	5 A A/4/@/@,',VVA A A A A A A A A A A A A A Aov~VGHO&&HO###99HO!!&"2v!FFHO##C((HO!!+)!DD		
 E 
MM.(%))VD\F:DV)WW     
II(( ?J*G  
      s   A  A$'A$c                    t          ddd          }t          j        t          j        t          ddd          t          j        d	d
| d          t          j        |ddd          dt          j        dd          t          j        dd          t          j        dd          d          }|	                                }t          d|                                j                   d S )Ntraining_iterationi     )	time_attrmax_tgrace_period   r   )cpugpu)	resourcesexpgGz?)r   r<   )namestopr   maxr   )	schedulermetricmodenum_samplesgMbP?g?g?    i   )threadsr   r   r   )
run_configtune_configparam_spacez!Best hyperparameters found were: )r   r   Tunerwith_resourcesr:   	RunConfig
TuneConfiguniformrandintr0   printget_best_resultr1   )num_training_iterationsschedtunerresultss       r9   
tune_mnistr^   =   s   #&c  E JK1Q3G3GHHH>#'?VWW
 
 
 O"	
 
 
 ,uc**S#..l2s++	
 
  E& iikkG	
-w/F/F/H/H/OPPPPP    __main__z--smoke-test
store_truezFinish quickly for testing)actionhelp   )num_cpusrA   i,  )rZ   )argparser"   sysfilelockr   rayr   ray.tune.schedulersr   version_infoexittensorflow.keras.datasetsr   ray.tune.integration.kerasr	   r:   r^   __name__ArgumentParserparseradd_argumentparse_known_argsargs_
smoke_testinit r_   r9   <module>ry      sh    				 



       



       7 7 7 7 7 7wCHQKKKK//////GGGGGG& & &RQ Q Q8 z$X$&&F
|2N     %%''GD! !JDO'DqqEEEEEE r_   