mirror of https://github.com/knative/docs.git
12 lines
367 B
Python
12 lines
367 B
Python
from bentoml import env, artifacts, api, BentoService
|
|
from bentoml.handlers import DataframeHandler
|
|
from bentoml.artifact import SklearnModelArtifact
|
|
|
|
@env(auto_pip_dependencies=True)
|
|
@artifacts([SklearnModelArtifact('model')])
|
|
class IrisClassifier(BentoService):
|
|
|
|
@api(DataframeHandler)
|
|
def predict(self, df):
|
|
return self.artifacts.model.predict(df)
|