merging to change repository base

4_train_keras_model.py

@@ -130,7 +130,7 @@ if __name__ == "__main__":
             pre_trained=True,
             freeze_layers=True,
             freeze_batch_norm=True,
-            base_model_type=ImageClassModels.EFFICIENTNET_V2B0,
+            base_model_type=ImageClassModels.EFFICIENTNET_V2S,
             dense_layer_neurons=1024,
             dropout_rate=.5,
         )

5_test_models.py

@@ -34,9 +34,7 @@ single_gen = ImageDataGenerator().flow_from_directory(
 
 
 for file in glob("./models/keras/*.hdf5"):
-
     print(file)
-    print(metrics_df["model"])
     if file in metrics_df.values:
         continue
     model = load_model(file)

all_model_output.csv

@@ -1,3 +1,5 @@
 model,test_acc,test_loss,single_acc,single_loss
 ./models/keras\pt-fl-fbn-efficientnet_v2b0-d1024-do0.5-l11.e-04-l21.e-04-5224-second_stage.hdf5,0.6720150708068079,1.7423864365349095,0.9893048128342246,0.4364729183409372
 ./models/keras\pt-fl-fbn-efficientnet_v2b0-d1024-do0.5-l11.e-04-l21.e-04-5224.hdf5,0.410029881772119,3.346152696366266,0.986096256684492,0.3234976000776315
+./models/keras\pt-fl-fbn-efficientnet_v2s-d1024-do0.5-l11.e-04-l21.e-04-8105-second_stage.hdf5,0.6850721060153306,1.675868156533777,0.9967914438502674,0.3373779159304851
+./models/keras\pt-fl-fbn-efficientnet_v2s-d1024-do0.5-l11.e-04-l21.e-04-8105.hdf5,0.37553592308691697,3.5500588697038067,0.9540106951871657,0.47270425785037834

modeling_utils/model_testing.py

@@ -1,4 +1,5 @@
 import numpy as np
+import pandas as pd
 from sklearn.metrics import accuracy_score, confusion_matrix, log_loss
 
 

print_all_wrong.py

@@ -1,4 +1,4 @@
 import pandas as pd
-df = pd.read_csv("models/keras/pt-fl-fbn-efficientnet_v2b0-d1024-do0.5-l11.e-04-l21.e-04-5224-second_stage.csv")
+df = pd.read_csv("models/keras/pt-fl-fbn-efficientnet_v2s-d1024-do0.5-l11.e-04-l21.e-04-8105-second_stage.csv")
 
 print(df.loc[df["prediction"] != df["true_val"]])