diff --git a/cli/main.py b/cli/main.py index c16648b..fcbf14c 100644 --- a/cli/main.py +++ b/cli/main.py @@ -2,12 +2,12 @@ Usage: spam -e - spam classifiers.manual.ManualClassifier - spam classifiers.features.FeatureClassifier - spam classifiers.manual.ManualClassifier -t 0.2 - spam classifiers.manual.ManualClassifier -a - spam classifiers.manual.ManualClassifier -a 5 - spam classifiers.manual.ManualClassifier -f 20 + spam models.manual.ManualClassifier + spam models.features.FeatureClassifier + spam models.manual.ManualClassifier -t 0.2 + spam models.manual.ManualClassifier -a + spam models.manual.ManualClassifier -a 5 + spam models.manual.ManualClassifier -f 20 """ import argparse @@ -32,7 +32,7 @@ def main(): parser.add_argument( "classifier", nargs="?", - help="Fully-qualified class, e.g. classifiers.manual.ManualClassifier", + help="Fully-qualified class, e.g. models.manual.ManualClassifier", ) parser.add_argument( "-e", "--explore", diff --git a/classifiers/__init__.py b/models/__init__.py similarity index 100% rename from classifiers/__init__.py rename to models/__init__.py diff --git a/classifiers/bow.py b/models/bow.py similarity index 95% rename from classifiers/bow.py rename to models/bow.py index 2174ab3..f48c610 100644 --- a/classifiers/bow.py +++ b/models/bow.py @@ -5,7 +5,7 @@ from sklearn.feature_extraction import DictVectorizer from sklearn.linear_model import LogisticRegression from sklearn.pipeline import Pipeline -from classifiers.cleaning import LowercaseTransformer, PunctuationRemover +from models.cleaning import LowercaseTransformer, PunctuationRemover class FeatureExtractor: diff --git a/classifiers/cleaning.py b/models/cleaning.py similarity index 100% rename from classifiers/cleaning.py rename to models/cleaning.py diff --git a/classifiers/features.py b/models/features.py similarity index 100% rename from classifiers/features.py rename to models/features.py diff --git a/classifiers/manual.py b/models/manual.py similarity index 100% rename from classifiers/manual.py rename to models/manual.py diff --git a/pyproject.toml b/pyproject.toml index f468aaf..e0be8b9 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -16,4 +16,4 @@ requires = ["hatchling"] build-backend = "hatchling.build" [tool.hatch.build.targets.wheel] -packages = ["cli", "classifiers"] +packages = ["cli", "models"]