Python model_interface Examples

Example #1

def test_with_serde(pd_model: Model):
    interface = model_interface(pd_model)

    obj = {'values': [{'a': 1, 'b': 1}]}

    data_type = pd_model.input_meta
    data = deserialize(obj, data_type)

    interface.predict(data)

Example #2

File: test_model_interface.py Project: zyfra/ebonite

def test_with_serde(pd_model: Model):
    interface = model_interface(pd_model)

    obj = {'values': [{'a': 1, 'b': 1}]}

    data_type, _ = pd_model.wrapper.method_signature('predict')
    data = deserialize(obj, data_type)

    interface.execute('predict', {'vector': data})

Example #3

File: test_model_runtime.py Project: zyfra/ebonite

 def load(self) -> Interface:
     mdl.wrapper.bind_model(MockModel(),
                            input_data=pd.DataFrame([[1, 0], [0, 1]],
                                                    columns=model_params))
     return model_interface(mdl)

Example #4

File: helpers.py Project: rubaha96/ebonite

 def load(self) -> Interface:
     model.ensure_loaded()
     return model_interface(model)

Example #5

def test_interface_types(pd_model: Model, data, prediction):
    interface = model_interface(pd_model)
    pred = interface.predict(data)
    assert (pred == prediction).all()

Example #6

File: test_model_interface.py Project: zyfra/ebonite

def test_interface_types(pd_model: Model, data, prediction):
    interface = model_interface(pd_model)
    assert interface.exposed_method_docs('predict') == pd_model.description
    pred = interface.execute('predict', {'vector': data})
    assert (pred == prediction).all()

Example #7

 def load(self) -> Interface:
     mdl.wrapper.bind_model(MockModel())
     return model_interface(mdl)

Example #8

File: test_model_interface.py Project: rubaha96/ebonite

def test_interface_types(pd_model: Model, data, prediction):
    interface = model_interface(pd_model)
    pred = interface.execute('predict', {'vector': data})
    assert (pred == prediction).all()