fix submission script lmao

script.py

@@ -71,13 +71,18 @@ def make_submission(test_metadata, model_path, model_name, output_csv_path="./su
         predictions.append(np.argmax(logits))
 
     test_metadata["class_id"] = predictions
-
+    
     user_pred_df = test_metadata.drop_duplicates("observation_id", keep="first")
+
+    for ix, row in user_pred_df.iterrows():
+        if row['class_id'] == 1604:
+            user_pred_df.loc[ix, 'class_id'] = -1
+
     user_pred_df[["observation_id", "class_id"]].to_csv(output_csv_path, index=None)
 
 def test_submission():
 
-    metadata_file_path = "../val_mini.csv"
+    metadata_file_path = "../trial_test.csv"
     test_metadata = pd.read_csv(metadata_file_path)
 
     make_submission(
@@ -90,18 +95,18 @@ def test_submission():
 
 if __name__ == "__main__":
 
-    # test_submission()
+    test_submission()
 
-    import zipfile
+    # import zipfile
 
-    with zipfile.ZipFile("/tmp/data/private_testset.zip", 'r') as zip_ref:
-        zip_ref.extractall("/tmp/data")
+    # with zipfile.ZipFile("/tmp/data/private_testset.zip", 'r') as zip_ref:
+    #     zip_ref.extractall("/tmp/data")
 
-    metadata_file_path = "./FungiCLEF2024_TestMetadata.csv"
-    test_metadata = pd.read_csv(metadata_file_path)
+    # metadata_file_path = "./FungiCLEF2024_TestMetadata.csv"
+    # test_metadata = pd.read_csv(metadata_file_path)
 
-    make_submission(
-        test_metadata=test_metadata,
-        model_path=MODEL_PATH,
-        model_name=MODEL_NAME
-    )
+    # make_submission(
+    #     test_metadata=test_metadata,
+    #     model_path=MODEL_PATH,
+    #     model_name=MODEL_NAME
+    # )