search

cancer-estimator / model

Model Search Repository
0 stars 0 forks source link

Add train flow of model search #41

Closed ryukinix closed 5 months ago

ryukinix commented 5 months ago

Example

pdm install
pdm run dvc pull
pdm run cancer-estimator-model
[train] start training model
[train] samples:  1309
[train] selected features(n=37):  AGE, SMOKING, YELLOW_FINGERS, ANXIETY, PEER_PRESSURE, CHRONIC_DISEASE, ALLERGY, WHEEZING, ALCOHOL_CONSUMING, FATIGUE, COUGHING, SHORTNESS_OF_BREATH, SWALLOWING_DIFFICULTY, CHEST_PAIN, GENDER_FEMALE, GENDER_MALE, COLD_SYMPTOMNS, RESPIRATORY_SYMPTOMNS, SNORING, AIR_POLLUTION, ALCOHOL_USE, DUST_ALLERGY, OCCUPATIONAL_HAZARDS, GENETIC_RISK, CHRONIC_LUNG_DISEASE, BALANCED_DIET, OBESITY, PASSIVE_SMOKER, COUGHING_OF_BLOOD, WEIGHT_LOSS, FREQUENT_COLD, DRY_COUGH, FEVER, PAINS, NASAL_CONGESTION, RUNNY_NOSE, DIARRHEA
0:      learn: 0.4642456        total: 48.3ms   remaining: 24.1s
100:    learn: 0.0951008        total: 201ms    remaining: 794ms
200:    learn: 0.0733792        total: 367ms    remaining: 547ms
300:    learn: 0.0561109        total: 543ms    remaining: 359ms
400:    learn: 0.0464717        total: 689ms    remaining: 170ms
499:    learn: 0.0381082        total: 838ms    remaining: 0us
[train] feature importance saved at: /home/lerax/Desktop/workspace/cancer-estimator-model/models/feature_importance.png
[train] model saved at: /home/lerax/Desktop/workspace/cancer-estimator-model/models/model.cbm

Closes #35