From d9f4676d59f7828ea3cb9960d7635723b05c5d46 Mon Sep 17 00:00:00 2001 From: atashnezhad Date: Fri, 30 Jun 2023 10:13:27 -0500 Subject: [PATCH] readme --- README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/README.md b/README.md index 3d39e98..85d8a2c 100644 --- a/README.md +++ b/README.md @@ -155,7 +155,7 @@ transfer_model = TransferModel( dataset_address=Path(__file__).parent / "dataset" ) -transfer_model.plot_classes_number() +transfer_model.plot_classes_number(figure_folder_path=Path(__file__).parent / "figures") transfer_model.analyze_image_names() transfer_model.plot_data_images(num_rows=3, num_cols=3) transfer_model.train_model()