Update dataset split for Zambia_CEO_2019

nasaharvest · May 26, 2023 · 1785602 · 1785602
1 parent 9188f90
commit 1785602
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/datasets.py b/datasets.py
@@ -811,7 +811,7 @@ def load_labels(self) -> pd.DataFrame:
                 filename="ceo-2019-Zambia-Cropland-(RCMRD-Set-1)-sample-data-2021-12-12.csv",
                 class_prob=lambda df: (df["Crop/non-crop"] == "Crop"),
                 start_year=2019,
-                train_val_test=(0.2, 0.4, 0.4),
+                train_val_test=(0.6, 0.2, 0.2),
                 latitude_col="lat",
                 longitude_col="lon",
                 filter_df=clean_ceo_data,
@@ -820,7 +820,7 @@ def load_labels(self) -> pd.DataFrame:
                 filename="ceo-2019-Zambia-Cropland-(RCMRD-Set-2)-sample-data-2021-12-12.csv",
                 class_prob=lambda df: (df["Crop/non-crop"] == "Crop"),
                 start_year=2019,
-                train_val_test=(0.2, 0.4, 0.4),
+                train_val_test=(0.6, 0.2, 0.2),
                 latitude_col="lat",
                 longitude_col="lon",
                 filter_df=clean_ceo_data,