Source code for tamr_unify_client.models.machine_learning_model

from tamr_unify_client.models.base_resource import BaseResource
from tamr_unify_client.models.operation import Operation


[docs]class MachineLearningModel(BaseResource): """A Unify Machine Learning model.""" @classmethod def from_json(cls, client, resource_json, api_path=None): return super().from_data(client, resource_json, api_path)
[docs] def train(self, **options): """Learn from verified labels. :param ``**options``: Options passed to underlying :class:`~tamr_unify_client.models.operation.Operation` . See :func:`~tamr_unify_client.models.operation.Operation.apply_options` . """ op_json = self.client.post(self.api_path + ":refresh").json() op = Operation.from_json(self.client, op_json) return op.apply_options(**options)
[docs] def predict(self, **options): """Suggest labels for unverified records. :param ``**options``: Options passed to underlying :class:`~tamr_unify_client.models.operation.Operation` . See :func:`~tamr_unify_client.models.operation.Operation.apply_options` . """ dependent_dataset = "/".join(self.api_path.split("/")[:-1]) op_json = self.client.post(dependent_dataset + ":refresh").json() op = Operation.from_json(self.client, op_json) return op.apply_options(**options)