Plotting and other updates - Githubissues

WorldCereal / presto-worldcereal

4 stars 0 forks source link

Plotting and other updates #41

Closed gabrieltseng closed 8 months ago

gabrieltseng commented 8 months ago

Plotting updates from #29 :

[x] Plot NDVI
[x] Plot probabilties for the sklearn models
[x] Ensure all plots use the same colorbar range

In addition:

[x] I have added 3 runs where we leave out different countries during finetuning, to better understand generalization effects. I also save those models' state dicts
[x] Fix a bug where we save the non-finetuned model state dict as the finetuned model

Model	Head	Overall F1	Latvia F1
WorldCereal	CatBoost	0.8489	0.8842
Full	CatBoost	0.8566	0.9051
No Latvia	CatBoost	0.8558	0.8694