| { | |
| "model_type": "tabpfn-finetuned", | |
| "base_model": "Prior-Labs/TabPFN-v2-clf", | |
| "task": "binary-classification", | |
| "domain": "species-distribution-modeling", | |
| "variants": { | |
| "nonspatial": { | |
| "filename": "tabpfn-sdm-nonspatial.pt", | |
| "description": "Standard train/test split evaluation", | |
| "step1_epochs": 40, | |
| "step2_epochs": 100, | |
| "final_val_roc_auc": 0.747, | |
| "final_val_pr_auc": 0.261, | |
| "final_train_loss": 0.460, | |
| "final_val_loss": 3.429 | |
| }, | |
| "spatial": { | |
| "filename": "tabpfn-sdm-spatial.pt", | |
| "description": "Spatially-separated (10km buffer) evaluation", | |
| "step1_epochs": 25, | |
| "step2_epochs": 50, | |
| "final_val_roc_auc": 0.653, | |
| "final_val_pr_auc": 0.144, | |
| "final_train_loss": 0.470, | |
| "final_val_loss": 5.469 | |
| } | |
| }, | |
| "training": { | |
| "optimizer": "AdamW", | |
| "lr_schedule": "OneCycleLR", | |
| "step1_learning_rate": 1e-5, | |
| "step2_learning_rate": 1e-6, | |
| "max_train_size_per_subbatch": 1500, | |
| "max_test_size": 1500, | |
| "n_estimators_training": 2, | |
| "n_estimators_inference": 8, | |
| "seed": 32639, | |
| "species_split_seed": 12345 | |
| }, | |
| "data": { | |
| "total_species": 226, | |
| "finetune_species_fraction": 0.625, | |
| "validation_species_fraction": 0.075, | |
| "test_species_fraction": 0.30, | |
| "regions": ["AWT", "NSW", "CAN", "NZ", "SA", "SWI"], | |
| "categorical_variables": { | |
| "CAN": ["ontveg"], | |
| "NSW": ["vegsys"], | |
| "NZ": ["age", "toxicats"], | |
| "SWI": ["calc"] | |
| } | |
| }, | |
| "checkpoint_format": { | |
| "keys": ["model_state_dict", "config", "history", "step1_path"], | |
| "framework": "pytorch" | |
| } | |
| } | |