o
    1 i                     @   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targetZreshaper   Zfit_transformr   )	Z	test_sizeZ	iris_dataxyencoderZtrain_xZtest_xZtrain_yZtest_y r   c/home/app/PaddleOCR-VL-test/.venv_paddleocr/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	threadingZ set_inter_op_parallelism_threadsZ set_intra_op_parallelism_threads)threadsr   r   r   set_keras_threads   s   r   )r   )
Z
tensorflowr   Zsklearn.datasetsr   Zsklearn.model_selectionr   Zsklearn.preprocessingr   r   r   r   r   r   r   <module>   s    

