
    &`i                     @    d dl Zd dlmZ d dlmZ d dlmZ ddZd Z	dS )	    N)	load_iris)train_test_split)OneHotEncoder皙?c                     t                      }|j        }|j                            dd          }t	          d          }|                    |          }t          ||          \  }}}}||||fS )N   F)sparse)r   datatargetreshaper   fit_transformr   )		test_size	iris_dataxyencodertrain_xtest_xtrain_ytest_ys	            k/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/ray/tune/examples/utils.pyget_iris_datar      st    IA  Q''A5)))Ga  A'71'='=$GVWfGVV++    c                     t           j        j                            |            t           j        j                            |            d S )N)tfconfig	threading set_inter_op_parallelism_threads set_intra_op_parallelism_threads)threadss    r   set_keras_threadsr"      s<     I88AAAI88AAAAAr   )r   )

tensorflowr   sklearn.datasetsr   sklearn.model_selectionr   sklearn.preprocessingr   r   r"    r   r   <module>r(      sy        & & & & & & 4 4 4 4 4 4 / / / / / /, , , ,B B B B Br   