readme

Atashnezhad · Jun 30, 2023 · d9f4676 · d9f4676
1 parent 575ea60
commit d9f4676
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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()