From e4070a681860e9901b038f19b706a5789ff209df Mon Sep 17 00:00:00 2001 From: adnankarol Date: Thu, 1 Aug 2024 20:07:16 +0200 Subject: [PATCH] Restructure --- __init__.py | 1 - classifierAgent/__init__.py | 0 classifierAgent.py => classifierAgent/classifierAgent.py | 2 +- setup.py | 4 ++-- 4 files changed, 3 insertions(+), 4 deletions(-) create mode 100644 classifierAgent/__init__.py rename classifierAgent.py => classifierAgent/classifierAgent.py (97%) diff --git a/__init__.py b/__init__.py index 9519c59..e69de29 100644 --- a/__init__.py +++ b/__init__.py @@ -1 +0,0 @@ -from classifierAgent import classifierAgent \ No newline at end of file diff --git a/classifierAgent/__init__.py b/classifierAgent/__init__.py new file mode 100644 index 0000000..e69de29 diff --git a/classifierAgent.py b/classifierAgent/classifierAgent.py similarity index 97% rename from classifierAgent.py rename to classifierAgent/classifierAgent.py index d5131e0..a10d003 100644 --- a/classifierAgent.py +++ b/classifierAgent/classifierAgent.py @@ -100,7 +100,7 @@ def plot_confusion_matrix(y_true, y_pred, model_name, show_plot=True): if show_plot: plt.show() -def classifierAgent(dataset, output_column, train_test_ratio, scaling_method=None, perform_grid_search=False, save_models=False, show_confusion_matrix=False, show_classification_report=False): +def classifier(dataset, output_column, train_test_ratio, scaling_method=None, perform_grid_search=False, save_models=False, show_confusion_matrix=False, show_classification_report=False): """ Trains multiple classifiers on the provided dataset and returns their performance metrics. diff --git a/setup.py b/setup.py index 0c2c519..1eb113e 100644 --- a/setup.py +++ b/setup.py @@ -18,11 +18,11 @@ def parse_requirements(filename): setup( name='classifierAgent', - version='1.0.8', + version='1.0.9', description='A Python package for performing classification on datasets.', long_description=long_description, long_description_content_type="text/markdown", - url='https://github.com/adnanmushtaq1996/ML-Classifier-Python-Package', + url='https://github.com/adnankarol/classifierAgent', author='Adnan Karol', author_email='adnanmushtaq5@gmail.com', license='MIT',