Metadata-Version: 2.1
Name: simplexfmrartifact
Version: 0.0.6
Summary: BentoML artifact framework for simpletransformers
Home-page: https://github.com/markmo/simplexfmrartifact
Author: Mark Moloney
Author-email: m4rkmo@gmail.com
License: UNKNOWN
Platform: UNKNOWN
Classifier: Programming Language :: Python :: 3
Classifier: License :: OSI Approved :: MIT License
Classifier: Operating System :: OS Independent
Requires-Python: >=3.6
Description-Content-Type: text/markdown
Requires-Dist: BentoML (==0.10.1)

# simplexfmrartifact

BentoML artifact framework for simpletransformers.

Installation:

    pip install simplexfmrartifact==0.0.4

Usage example (decorate service):

    from simplexfmrartifact.simpletransformers import SimpleTransformersModelArtifact

    @artifacts([SimpleTransformersModelArtifact('tm_train3_roberta_l_weigh')])
    class MyBentoService(BentoService):


Usage example (package model):

    svc = MyBentoService()

    opts = {
        'classpackage': 'simpletransformers.classification',
        'classname': 'MultiLabelClassificationModel',
        'num_labels': 33,
        'args': {
            'use_multiprocessing': False,
            'silent': True
        }
    }

    svc.pack('tm_train3_roberta_l_weigh', model_path, opts)

Alternatively, during training:

    svc.pack('tm_train3_roberta_l_weigh', {'model': my_trained_model})


