example-split-with-submodels.py

from yhat import Yhat, YhatModel
import numpy as np


class SplitTestExample(YhatModel):
    def execute(self, data):
        # randomly select model to use. `p` defines the split percentage
        endpoints = ["ModelA", "ModelB"]
        endpoint = np.random.choice(endpoints, p=[0.2, 0.8])
        # make prediction to a "sub-model". this will give you back the full API response
        response = yh.predict(endpoint, data)
        # format the results and return. we're only going to grab the 'result' value
        # from the API response
        output = response['result']
        # assign a label for which endpoint/variant was used, and then return the output
        output['variant'] = endpoint
        return output

yh = Yhat(USERNAME, APIKEY, URL)
yh.deploy("SplitTestSuperModel", SplitTestExample, globals())