crowncode-backend / app /training /train_classifier.py

Commit History

feat: define families dictionary for candidate classifiers in _build_candidate_families function
1d86705

Rthur2003 commited on

feat: enhance candidate selection with class ratio and calibrated SVC for improved model performance
57f19bf

Rthur2003 commited on

feat: implement optimal threshold calculation for binary predictions in train function
058eadc

Rthur2003 commited on

feat: add CalibratedClassifierCV and roc_curve imports for enhanced model evaluation
27cd744

Rthur2003 commited on

Add Gradio dependency to requirements.txt with version constraints
20fe6c3

Rthur2003 commited on

feat: train all models on full data and save their paths
94ed8e9

Rthur2003 commited on

feat: update model parameters for Random Forest, Gradient Boosting, XGBoost, and LightGBM to enhance performance
e68824c

Rthur2003 commited on

feat: adjust Random Forest parameters for improved model performance
905436c

Rthur2003 commited on

feat: optimize Gradient Boosting and XGBoost parameters for improved model performance
0d5b4e8

Rthur2003 commited on

feat: update model parameters and fix data leakage by removing duration_sec and sample_rate from features
bb6655d

Rthur2003 commited on

feat: exclude additional metadata columns from feature loading to prevent data leakage
4b87da0

Rthur2003 commited on

feat: enhance training pipeline with multi-model evaluation and detailed metrics
5446f0d

Rthur2003 commited on

feat: add AURIS classifier training module with model evaluation and feature importance
337d9ae

Rthur2003 commited on