Import your own SK-Learn models
Deploy-ML aims to make the process of training, testing and deploying machine learning easier and simplier. However, sometimes the model required is not directly supported, or a collegue has already defined a model. The import class is for users who define their own sk-learn model, but want to use the package for training and deployment reasons. To use your sk-learn model, import the import class:
from deployml.sklearn.models.import import ImportBase
- model = Sk-Learn model that you have imported and defined
- model_title = string describing the title of the model for deployment package
Below is an example of importing a logistic regression model from sk-learn, and using the deploy-ml import class:
from sklearn.linear_model import LogisticRegression from deployml.sklearn.models.import import ImportBase imported_model = LogisticRegression(penalty=’l2’, dual=False, tol=0.0001, C=1.0, fit_intercept=True, intercept_scaling=1, class_weight=None, random_state=None, solver=’liblinear’, max_iter=100, multi_class=’ovr’, verbose=0, warm_start=False, n_jobs=1) model = ImportBase(model=imported_model, model_title="Logistic Regression")
Now the imported model has all the training and deployment functions of deploy-ml.