Metadata-Version: 2.1
Name: simplexfmrartifact
Version: 0.0.12
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

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()

    metadata = {
        "classpackage": "simpletransformers.classification", 
        "classname": "ClassificationModel", 
        "opts": {
            "use_cuda": true, 
            "num_labels": 2, 
            "args": {
                "use_multiprocessing": false, 
                "silent": true, 
                "eval_batch_size": 10, 
                "fp16": false
            }
        }
    }

    svc.pack(model_name, model_path, metadata)

Alternatively, during training:

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


