Skip to content

Commit

Permalink
Update entry point
Browse files Browse the repository at this point in the history
  • Loading branch information
adnankarol authored and adnankarol committed Aug 1, 2024
1 parent c2eaa6e commit c72d902
Show file tree
Hide file tree
Showing 4 changed files with 36 additions and 29 deletions.
2 changes: 1 addition & 1 deletion classifierAgent/__init__.py
Original file line number Diff line number Diff line change
@@ -1 +1 @@
from .classifierAgent import classifier
from .classifierAgent import classifierAgent
4 changes: 2 additions & 2 deletions classifierAgent/classifierAgent.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
__author__ = "Adnan Karol"
__version__ = "1.0.7"
__version__ = "1.1.1"
__maintainer__ = "Adnan Karol"
__email__ = "adnanmushtaq5@gmail.com"
__status__ = "PROD"
Expand Down Expand Up @@ -100,7 +100,7 @@ def plot_confusion_matrix(y_true, y_pred, model_name, show_plot=True):
if show_plot:
plt.show()

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):
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):
"""
Trains multiple classifiers on the provided dataset and returns their performance metrics.
Expand Down
11 changes: 8 additions & 3 deletions setup.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
__author__ = "Adnan Karol"
__version__ = "1.0.7"
__version__ = "1.1.1"
__maintainer__ = "Adnan Karol"
__email__ = "adnanmushtaq5@gmail.com"
__status__ = "PROD"
Expand All @@ -18,7 +18,7 @@ def parse_requirements(filename):

setup(
name='classifierAgent',
version='1.1.0',
version='1.1.1',
description='A Python package for performing classification on datasets.',
long_description=long_description,
long_description_content_type="text/markdown",
Expand All @@ -37,5 +37,10 @@ def parse_requirements(filename):
],
python_requires='>=3.10',
packages=find_packages(),
install_requires=parse_requirements('requirements.txt')
install_requires=parse_requirements('requirements.txt'),
entry_points={
'console_scripts': [
'classifierAgent=classifierAgent:classifierAgent',
],
},
)
Loading

0 comments on commit c72d902

Please sign in to comment.