auris-models / training_results.json
Rthur2003's picture
Upload training_results.json with huggingface_hub
725537c verified
Raw History Blame Contribute Delete
6 kB
{
"Logistic Regression": {
"accuracy": 0.7779,
"precision": 0.6983,
"recall": 0.7848,
"f1": 0.739,
"roc_auc": 0.8511,
"validation_auc": 0.8383,
"selection_time_sec": 0.21,
"train_time_sec": 0.3,
"selected_params": {
"C": 2.0,
"class_weight": "balanced",
"max_iter": 2500
},
"optimal_threshold": 0.5
},
"Random Forest": {
"accuracy": 0.8604,
"precision": 0.8554,
"recall": 0.7843,
"f1": 0.8183,
"roc_auc": 0.9393,
"validation_auc": 0.9302,
"selection_time_sec": 4.74,
"train_time_sec": 9.4,
"selected_params": {
"n_estimators": 500,
"max_depth": null,
"min_samples_leaf": 1,
"min_samples_split": 2,
"class_weight": "balanced_subsample",
"max_features": "log2"
},
"optimal_threshold": 0.5
},
"Gradient Boosting": {
"accuracy": 0.8685,
"precision": 0.8454,
"recall": 0.8223,
"f1": 0.8337,
"roc_auc": 0.9406,
"validation_auc": 0.9297,
"selection_time_sec": 18.64,
"train_time_sec": 33.84,
"selected_params": {
"n_estimators": 180,
"max_depth": 4,
"learning_rate": 0.07,
"subsample": 0.75,
"min_samples_leaf": 8,
"min_samples_split": 16
},
"optimal_threshold": 0.5
},
"SVM (RBF)": {
"accuracy": 0.8612,
"precision": 0.8331,
"recall": 0.8175,
"f1": 0.8252,
"roc_auc": 0.9347,
"validation_auc": 0.9336,
"selection_time_sec": 13.73,
"train_time_sec": 19.79,
"selected_params": {
"C": 10.0,
"gamma": 0.05,
"class_weight": "balanced"
},
"optimal_threshold": 0.5
},
"MLP Neural Network": {
"accuracy": 0.8545,
"precision": 0.8169,
"recall": 0.8208,
"f1": 0.8189,
"roc_auc": 0.9258,
"validation_auc": 0.9278,
"selection_time_sec": 4.72,
"train_time_sec": 7.49,
"selected_params": {
"hidden_layer_sizes": [
192,
96,
32
],
"alpha": 0.001,
"max_iter": 600,
"validation_fraction": 0.15
},
"optimal_threshold": 0.5
},
"XGBoost": {
"accuracy": 0.8735,
"precision": 0.8512,
"recall": 0.8295,
"f1": 0.8402,
"roc_auc": 0.9463,
"validation_auc": 0.9376,
"selection_time_sec": 1.44,
"train_time_sec": 2.1,
"selected_params": {
"n_estimators": 240,
"max_depth": 5,
"learning_rate": 0.06,
"subsample": 0.75,
"colsample_bytree": 0.75,
"min_child_weight": 6,
"reg_alpha": 0.4,
"reg_lambda": 1.5,
"gamma": 0.2
},
"optimal_threshold": 0.5
},
"LightGBM": {
"accuracy": 0.8839,
"precision": 0.8441,
"recall": 0.8713,
"f1": 0.8575,
"roc_auc": 0.9549,
"validation_auc": 0.9472,
"selection_time_sec": 3.25,
"train_time_sec": 2.95,
"selected_params": {
"n_estimators": 300,
"max_depth": -1,
"learning_rate": 0.05,
"num_leaves": 31,
"subsample": 0.8,
"colsample_bytree": 0.8,
"min_child_samples": 20,
"reg_alpha": 0.1,
"reg_lambda": 1.0
},
"optimal_threshold": 0.431577
},
"_best_model": "LightGBM",
"_n_samples": 5195,
"_n_features": 47,
"_n_folds": 5,
"_dataset_path": "d:\\CrownCode\\DataSet\\features.csv",
"_class_balance": {
"ai": 2082,
"human": 3113
},
"_data_leakage_fix": "duration_sec and sample_rate removed from features; scaler fitted per fold during CV",
"_model_paths": {
"Logistic Regression": "d:\\CrownCode\\hf-crowncode-backend\\models\\model_logistic_regression.pkl",
"Random Forest": "d:\\CrownCode\\hf-crowncode-backend\\models\\model_random_forest.pkl",
"Gradient Boosting": "d:\\CrownCode\\hf-crowncode-backend\\models\\model_gradient_boosting.pkl",
"SVM (RBF)": "d:\\CrownCode\\hf-crowncode-backend\\models\\model_svm_rbf.pkl",
"MLP Neural Network": "d:\\CrownCode\\hf-crowncode-backend\\models\\model_mlp_neural_network.pkl",
"XGBoost": "d:\\CrownCode\\hf-crowncode-backend\\models\\model_xgboost.pkl",
"LightGBM": "d:\\CrownCode\\hf-crowncode-backend\\models\\model_lightgbm.pkl"
},
"_feature_importance": {
"spectral_flatness_std": 0.061889,
"spectral_contrast_mean": 0.046667,
"rms_energy": 0.045556,
"onset_strength_std": 0.038778,
"spectral_flatness_mean": 0.037,
"rms_dynamic_range": 0.034556,
"onset_strength_mean": 0.033222,
"rms_std": 0.029778,
"beat_count": 0.029778,
"mfcc_delta_var": 0.028889,
"chroma_std": 0.028111,
"mfcc_variance": 0.027889,
"tonnetz_std": 0.027333,
"chroma_entropy": 0.022333,
"tempo_cv": 0.022222,
"zero_crossing_rate": 0.021333,
"spectral_bandwidth_std": 0.020444,
"breath_pattern_score": 0.020444,
"mel_flatness": 0.020333,
"spectral_centroid_mean": 0.019222,
"formant_consistency_score": 0.019222,
"vocal_energy_ratio": 0.019111,
"harmonic_ratio": 0.017889,
"zero_crossing_std": 0.017556,
"pitch_std_cents": 0.017333,
"harmonic_structure": 0.017222,
"vocal_harmonic_ratio": 0.017222,
"tempo_stability": 0.016444,
"chroma_transition_rate": 0.016444,
"spectral_contrast_std": 0.016333,
"pitch_mean_hz": 0.016222,
"spectral_regularity": 0.015889,
"vibrato_regularity_score": 0.015889,
"vibrato_extent_cents": 0.015889,
"spectral_bandwidth_mean": 0.014667,
"spectral_rolloff_std": 0.014,
"tempo_bpm": 0.014,
"temporal_patterns": 0.013556,
"mfcc_delta2_var": 0.013111,
"vocal_ai_score": 0.012667,
"spectral_centroid_std": 0.012556,
"pitch_stability_score": 0.011556,
"spectral_rolloff_mean": 0.010889,
"vibrato_rate_hz": 0.010333,
"vocal_texture_score": 0.009556,
"vocal_confidence": 0.008667,
"has_vocals": 0.0
}
}