"""
UT Sample Class
============================================
"""
from .sample import Sample
from ..field import TextField
__all__ = ['UTSample']
[docs]class UTSample(Sample):
r""" Universal Transformation sample.
Universe Transformation is not a subtask of NLP,
implemented for providing universal text transformation function.
"""
def __init__(
self,
data,
origin=None,
sample_id=None
):
self.x = None
super().__init__(data, origin=origin, sample_id=sample_id)
def __repr__(self):
return 'UTSample'
def check_data(self, data):
assert 'x' in data and isinstance(data['x'], str)
[docs] def load(self, data):
r"""
Convert data dict which contains essential information to SASample.
:param dict data: contains 'x' key at least.
:return:
"""
self.x = TextField(data['x'])
def dump(self):
return {'x': self.x.text, 'sample_id': self.sample_id}