o
    ci                     @   sB   d dl Zd dlmZ d dlmZ d dlmZ d
ddZdd	 Z	dS )    N)	load_iris)train_test_split)OneHotEncoder皙?c           	      C   sL   t  }|j}|jdd}td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_y r   K/home/ubuntu/.local/lib/python3.10/site-packages/ray/tune/examples/utils.pyget_iris_data   s   

r   c                 C   s    t jj|  t jj|  d S )N)tfconfig	threading set_inter_op_parallelism_threads set_intra_op_parallelism_threads)threadsr   r   r   set_keras_threads   s   r   )r   )

tensorflowr   sklearn.datasetsr   sklearn.model_selectionr   sklearn.preprocessingr   r   r   r   r   r   r   <module>   s    

