fix: extend xlim in plot_ml_vs_dl to prevent clipping of value labels 4744538 Rthur2003 commited on 12 days ago
fix: update calibration and precision-recall plotting functions to load data from files 129f521 Rthur2003 commited on 14 days ago
fix: update confusion matrix and score distribution functions to load pre-computed CV predictions fa18330 Rthur2003 commited on 14 days ago
feat: add script to generate publication-ready figures for AURIS paper e8077a2 Rthur2003 commited on 14 days ago
fix: update evaluation to use optimal threshold and adjust BCEWithLogitsLoss for class imbalance 9b6c85d Rthur2003 commited on 28 days ago
fix: adjust BCEWithLogitsLoss to handle class imbalance with pos_weight 5ac76c0 Rthur2003 commited on 28 days ago
feat: define families dictionary for candidate classifiers in _build_candidate_families function 1d86705 Rthur2003 commited on 28 days ago
feat: enhance candidate selection with class ratio and calibrated SVC for improved model performance 57f19bf Rthur2003 commited on 28 days ago
feat: implement optimal threshold calculation for binary predictions in train function 058eadc Rthur2003 commited on 28 days ago
feat: add CalibratedClassifierCV and roc_curve imports for enhanced model evaluation 27cd744 Rthur2003 commited on 28 days ago
feat: add TorchSklearnWrapper for PyTorch model compatibility with sklearn and enhance final model training process bb33480 Rthur2003 commited on Apr 24
Add Gradio dependency to requirements.txt with version constraints 20fe6c3 Rthur2003 commited on Apr 16
feat: implement mixed precision training and gradient accumulation in training loop 076d979 Rthur2003 commited on Apr 16
feat: adjust confidence range in analysis result and optimize parallel processing for feature extraction, vocal analysis, and CLAP detection 8e5e154 Rthur2003 commited on Apr 16
feat: update model parameters for Random Forest, Gradient Boosting, XGBoost, and LightGBM to enhance performance e68824c Rthur2003 commited on Apr 16
feat: adjust Random Forest parameters for improved model performance 905436c Rthur2003 commited on Apr 16
feat: optimize Gradient Boosting and XGBoost parameters for improved model performance 0d5b4e8 Rthur2003 commited on Apr 16
feat: refactor training history figure generation to use GradientBoostingClassifier and improve metric tracking 1c2de42 Rthur2003 commited on Apr 16
feat: update model parameters and fix data leakage by removing duration_sec and sample_rate from features bb6655d Rthur2003 commited on Apr 16
feat: add deep learning classifier comparison script for AURIS with multiple architectures b8d143b Rthur2003 commited on Apr 16
feat: update excluded columns in feature statistics computation to prevent data leakage edc551d Rthur2003 commited on Apr 16
feat: exclude additional metadata columns from feature loading to prevent data leakage 4b87da0 Rthur2003 commited on Apr 16
feat: update output directory for generated figures to docs/academic/figures 983f3ef Rthur2003 commited on Apr 15
feat: add deep-learning figure generation script for enhanced model evaluation 917199d Rthur2003 commited on Apr 15
feat: enhance calibration figure with Brier score and update prediction functions 255af12 Rthur2003 commited on Apr 15
feat: update ROC and Precision-Recall curve functions to include best model visualization ca749ec Rthur2003 commited on Apr 15
feat: implement cross-validated predictions caching and update confusion matrix function 0ea8f54 Rthur2003 commited on Apr 15
feat: add additional imports for model evaluation and cross-validation 853f480 Rthur2003 commited on Apr 15
feat: add figure generation script for training results visualization 94e94a1 Rthur2003 commited on Apr 15
feat: add feature statistics computation from training CSV for enhanced analysis 575dde4 Rthur2003 commited on Apr 15
feat: implement resume functionality in batch feature extraction to skip processed samples 8f111ee Rthur2003 commited on Apr 15
feat: enhance visualization pipeline with additional figures and manifest support b06ac73 Rthur2003 commited on Apr 14
feat: add dataset statistics and class balance visualizations to enhance analysis capabilities a04f1c8 Rthur2003 commited on Apr 14
feat: refactor worker function for batch feature extraction to support multiprocessing 2a59987 Rthur2003 commited on Apr 14
feat: implement parallel processing for batch feature extraction to improve performance 282a605 Rthur2003 commited on Apr 14
feat: enhance training pipeline with multi-model evaluation and detailed metrics 5446f0d Rthur2003 commited on Mar 31
feat: enhance feature extraction with additional spectral and temporal metrics 7f1ed48 Rthur2003 commited on Mar 31
fix: gitignore models/data and add logging to feature extraction d33093c Rthur2003 Claude Opus 4.6 commited on Mar 31
fix: add debug logging for vocal extraction failures in extract_sample_features 91e078e Rthur2003 commited on Mar 28
fix: add return type annotations and docstrings for clarity in Wav2Vec2MusicClassifier and AudioDataset e93262a Rthur2003 commited on Mar 28
feat: add wav2vec2 classifier for AI music detection with training loop and dataset handling 7dc37a8 Rthur2003 commited on Mar 28
feat: add AURIS classifier training module with model evaluation and feature importance 337d9ae Rthur2003 commited on Mar 28
feat: implement batch feature extraction for AURIS training pipeline b74a8cb Rthur2003 commited on Mar 28
feat: add evaluation framework for AURIS models with metrics computation bc1975b Rthur2003 commited on Mar 28
feat: add AURIS training pipeline module for dataset loading and model training 621b77e Rthur2003 commited on Mar 28