Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Make a ClassificationPredictor class for seamless predictions from various models #66

Open
2 tasks
abhi8893 opened this issue Jun 16, 2021 · 0 comments
Open
2 tasks
Assignees
Labels
feature Typically a new function or a class
Projects

Comments

@abhi8893
Copy link
Owner

class ClassificationPredictor:

    def __init__(self, model, class_mode, class_names=None, featurizer=None):
        self.model = model
        self.featurizer = featurizer
        self.class_mode = class_mode

    def predict(self, X, featurize=False):
        pass

    def predict_proba(self, X, featurize=False):
        pass

Features:

  • Add a featurizer option
  • Model can be a simple predict proba function, or a sklearn model, or a tf.keras model (Yet to decide how to handle this)
@abhi8893 abhi8893 added the feature Typically a new function or a class label Jun 16, 2021
@abhi8893 abhi8893 self-assigned this Jun 16, 2021
@abhi8893 abhi8893 added this to Backlog in Workload via automation Jun 16, 2021
@abhi8893 abhi8893 changed the title Make a 1ClassificationPredictor` class for seamless predictions from various models Make a ClassificationPredictor class for seamless predictions from various models Jun 22, 2021
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
feature Typically a new function or a class
Projects
Workload
Backlog
Development

No branches or pull requests

1 participant