crowncode-backend / app /training

Commit History

fix: extend xlim in plot_ml_vs_dl to prevent clipping of value labels
4744538

Rthur2003 commited on

fix: update calibration and precision-recall plotting functions to load data from files
129f521

Rthur2003 commited on

fix: update confusion matrix and score distribution functions to load pre-computed CV predictions
fa18330

Rthur2003 commited on

feat: add script to generate publication-ready figures for AURIS paper
e8077a2

Rthur2003 commited on

fix: update evaluation to use optimal threshold and adjust BCEWithLogitsLoss for class imbalance
9b6c85d

Rthur2003 commited on

fix: adjust BCEWithLogitsLoss to handle class imbalance with pos_weight
5ac76c0

Rthur2003 commited on

fix: add missing roc_curve metric import for evaluation
bfbdc0b

Rthur2003 commited on

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

feat: add TorchSklearnWrapper for PyTorch model compatibility with sklearn and enhance final model training process
bb33480

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: implement mixed precision training and gradient accumulation in training loop
076d979

Rthur2003 commited on

feat: adjust confidence range in analysis result and optimize parallel processing for feature extraction, vocal analysis, and CLAP detection
8e5e154

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: refactor training history figure generation to use GradientBoostingClassifier and improve metric tracking
1c2de42

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: add deep learning classifier comparison script for AURIS with multiple architectures
b8d143b

Rthur2003 commited on

feat: update excluded columns in feature statistics computation to prevent data leakage
edc551d

Rthur2003 commited on

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

Rthur2003 commited on

feat: update output directory for generated figures to docs/academic/figures
983f3ef

Rthur2003 commited on

feat: add deep-learning figure generation script for enhanced model evaluation
917199d

Rthur2003 commited on

feat: enhance calibration figure with Brier score and update prediction functions
255af12

Rthur2003 commited on

feat: update ROC and Precision-Recall curve functions to include best model visualization
ca749ec

Rthur2003 commited on

feat: implement cross-validated predictions caching and update confusion matrix function
0ea8f54

Rthur2003 commited on

feat: add additional imports for model evaluation and cross-validation
853f480

Rthur2003 commited on

feat: add figure generation script for training results visualization
94e94a1

Rthur2003 commited on

fix: rename variable for clarity in compute_stats function
7e3e9a6

Rthur2003 commited on

feat: add feature statistics computation from training CSV for enhanced analysis
575dde4

Rthur2003 commited on

feat: implement resume functionality in batch feature extraction to skip processed samples
8f111ee

Rthur2003 commited on

feat: enhance visualization pipeline with additional figures and manifest support
b06ac73

Rthur2003 commited on

feat: add dataset statistics and class balance visualizations to enhance analysis capabilities
a04f1c8

Rthur2003 commited on

feat: refactor worker function for batch feature extraction to support multiprocessing
2a59987

Rthur2003 commited on

feat: implement parallel processing for batch feature extraction to improve performance
282a605

Rthur2003 commited on

feat: add full training and evaluation pipeline for AURIS
36a8a94

Rthur2003 commited on

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

Rthur2003 commited on

feat: enhance feature extraction with additional spectral and temporal metrics
7f1ed48

Rthur2003 commited on

fix: gitignore models/data and add logging to feature extraction
d33093c

Rthur2003 Claude Opus 4.6 commited on

fix: add debug logging for vocal extraction failures in extract_sample_features
91e078e

Rthur2003 commited on

fix: add return type annotations and docstrings for clarity in Wav2Vec2MusicClassifier and AudioDataset
e93262a

Rthur2003 commited on

feat: add wav2vec2 classifier for AI music detection with training loop and dataset handling
7dc37a8

Rthur2003 commited on

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

Rthur2003 commited on

feat: implement batch feature extraction for AURIS training pipeline
b74a8cb

Rthur2003 commited on

feat: add evaluation framework for AURIS models with metrics computation
bc1975b

Rthur2003 commited on

feat: add SONICS dataset loader for AURIS training pipeline
bfbcec4

Rthur2003 commited on

feat: add AURIS training pipeline module for dataset loading and model training
621b77e

Rthur2003 commited on