| { |
| "model_name": "Mouse_USV_Detector_ProSAP1_Shank2", |
| "model_type": "USVDetectorCNN", |
| "task": "audio-classification", |
| "protocol": "S2-4-65 ProSAP1/Shank2 Male-Oestrus Female Interactions", |
| "architecture": { |
| "input_size": [64, 64], |
| "num_classes": 2, |
| "classes": ["noise", "usv"] |
| }, |
| "audio_preprocessing": { |
| "sample_rate": 250000, |
| "nfft": 0.0032, |
| "overlap": 0.0028, |
| "hop_length_samples": 800, |
| "freq_range": [40000, 100000], |
| "freq_range_description": "Mouse USV frequency range (40-100 kHz)" |
| }, |
| "training": { |
| "dataset_size": 9334, |
| "train_samples": 7188, |
| "val_samples": 2146, |
| "num_epochs": 10, |
| "batch_size": 32, |
| "learning_rate": 0.001, |
| "optimizer": "Adam", |
| "loss_function": "CrossEntropyLoss" |
| }, |
| "performance": { |
| "validation_accuracy": 96.0, |
| "noise_accuracy": 94.7, |
| "usv_accuracy": 98.0 |
| }, |
| "framework": "pytorch", |
| "pytorch_version": "2.0+", |
| "model_size_mb": 41.9, |
| "parameters": 10467202 |
| } |
