From f70690efa8276cefcf3e8ccd3e3e367f259e537a Mon Sep 17 00:00:00 2001 From: Lucas Oskorep Date: Tue, 7 Jun 2022 19:33:40 -0400 Subject: [PATCH] merging to change repository base --- 4_train_keras_model.py | 2 +- 5_test_models.py | 2 -- all_model_output.csv | 2 ++ modeling_utils/model_testing.py | 1 + print_all_wrong.py | 2 +- 5 files changed, 5 insertions(+), 4 deletions(-) diff --git a/4_train_keras_model.py b/4_train_keras_model.py index 795f7ec..702c873 100644 --- a/4_train_keras_model.py +++ b/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, ) diff --git a/5_test_models.py b/5_test_models.py index 5ad9fd3..8700bf0 100644 --- a/5_test_models.py +++ b/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) diff --git a/all_model_output.csv b/all_model_output.csv index 9753e63..43f766e 100644 --- a/all_model_output.csv +++ b/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 diff --git a/modeling_utils/model_testing.py b/modeling_utils/model_testing.py index b03f39a..3922dd6 100644 --- a/modeling_utils/model_testing.py +++ b/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 diff --git a/print_all_wrong.py b/print_all_wrong.py index 439b175..eb8a830 100644 --- a/print_all_wrong.py +++ b/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"]])