WorldCereal / presto-worldcereal

2 stars 0 forks source link

Update code to work with datasets shared in #34 #38

Closed gabrieltseng closed 5 months ago

gabrieltseng commented 5 months ago

Run: https://wandb.ai/nasa-harvest/presto-worldcereal/runs/zwolm9c6. Results with all the data (from this run):

Model F1 score Precision Recall
CatBoost 0.849 0.846 0.852
Presto finetuning 0.857 0.813 0.907
Presto finetuning + RF 0.866 0.894 0.839
Presto finetuning + LR 0.859 0.812 0.911
Presto finetuning + CatBoost 0.863 0.822 0.908
kvantricht commented 5 months ago

@gabrieltseng Note that I uploaded a new file to the OneDrive containing the list of sample_ids that contain the original VAL set. You could use those to replicate the previous split.