o
    1 i:                  	   @   s  d dl Zd dlZd dlZd dlmZ d dlmZ d dl	m
Z
 d dlmZ defddZdefd	d
Zedkrd dlZe Zejdddd e \ZZdddgdeddgeddedddZejejskeneejddde dedZe  Z!e"de!# j dS dS )     N)train_test_split)tune)TuneReportCheckpointCallback)ASHASchedulerconfigc           	   	   C   sl   t jjdd\}}t||dd\}}}}tj||d}tj||d}tj| ||gdgtddd	gd
 d S )NTZ
return_X_yg      ?)Z	test_sizelabelevalzeval-binary_errorzeval-binary_logloss)binary_errorbinary_logloss)Z
valid_setsZvalid_names	callbacks)sklearndatasetsload_breast_cancerr   lgbDatasettrainr   )	r   datatargetZtrain_xZtest_xZtrain_yZtest_y	train_setZtest_set r   n/home/app/PaddleOCR-VL-test/.venv_paddleocr/lib/python3.10/site-packages/ray/tune/examples/lightgbm_example.pytrain_breast_cancer   s    
r   c              
   C   sJ   t jjdd\}}tj||d}tj| |dtdddddd	d
gd d S )NTr   r   zvalid-binary_error-meanzvalid-binary_logloss-meanzvalid-binary_error-stdvzvalid-binary_logloss-stdv)r   r   Zbinary_error_stdvZbinary_logloss_stdvr   )Z	frequency)Z
stratifiedr   )r   r   r   r   r   Zcvr   )r   r   r   r   r   r   r   train_breast_cancer_cv)   s    
r   __main__z--use-cv
store_truez$Use `lgb.cv` instead of `lgb.train`.)actionhelpbinaryr   r   Zgbdtdart
   i  g:0yE>g?)Z	objectivemetricverboseZboosting_typeZ
num_leavesZlearning_ratemin   )r#   modeZnum_samplesZ	scheduler)Ztune_configZparam_spacez!Best hyperparameters found were: )$Zlightgbmr   Zsklearn.datasetsr   Zsklearn.metricsZsklearn.model_selectionr   Zrayr   Zray.tune.integration.lightgbmr   Zray.tune.schedulersr   dictr   r   __name__argparseArgumentParserparseradd_argumentparse_known_argsargs_Zgrid_searchrandintZ
loguniformr   ZTunerZuse_cvZ
TuneConfigZtunerfitresultsprintZget_best_resultr   r   r   r   <module>   sH     

	
