Sintaxe
newrelic.agent.wrap_mlmodel(model, name=None, version=None, feature_names=None, label_names=None, metadata=None)
Permite instrumentação manual de modelos de aprendizado de máquina.
Requisitos
Agente Python versão 9.1.0 ou mais alto.
Descrição
Isso permite a instrumentação manual de modelos de aprendizado de máquina.
Parâmetro
Parâmetro | Descrição |
---|---|
objeto | Obrigatório. |
corda | Opcional. O nome do modelo personalizado. |
corda | Opcional. A versão de lançamento do modelo personalizado. |
lista de string | Opcional. Uma lista de strings que denotam o(s) nome(s) do(s) recurso(s). |
lista de string | Opcional. Uma lista de strings que denotam os nomes dos rótulos. |
dict | Opcional. metadados para anexar ao modelo. |
Valores de retorno
Nenhum.
Exemplos
Envolver modelo de aprendizado de máquina
Um exemplo de instrumentado um modelo de aprendizado de máquina personalizado:
def wrap_ml_example(): x_train = [[0, 0], [1, 1]] y_train = [0, 1] x_test = [[1.0, 2.0]]
model = CustomTestModel().fit(x_train, y_train) wrap_mlmodel( model, name="MyCustomModel", version="1.2.3", feature=["feature0", "feature1"], label=["label0"], metadata={"metadata1": "value1", "metadata2": "value2"}, )
labels = model.predict(x_test)
return model