Source code for tamr_unify_client.base_model

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


[docs]class MachineLearningModel(BaseResource): """A Tamr 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.operation.Operation` . See :func:`~tamr_unify_client.operation.Operation.apply_options` . :returns: The resultant operation. :rtype: :class:`~tamr_unify_client.operation.Operation` """ op_json = self.client.post(self.api_path + ":refresh").successful().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.operation.Operation` . See :func:`~tamr_unify_client.operation.Operation.apply_options` . :returns: The resultant operation. :rtype: :class:`~tamr_unify_client.operation.Operation` """ dependent_dataset = "/".join(self.api_path.split("/")[:-1]) op_json = self.client.post(dependent_dataset + ":refresh").successful().json() op = Operation.from_json(self.client, op_json) return op.apply_options(**options)
def __repr__(self): return ( f"{self.__class__.__module__}." f"{self.__class__.__qualname__}(" f"relative_id={self.relative_id!r})" )