from .. import DataWrapper
[docs]class HeterogeneousEmbeddingDataWrapper(DataWrapper):
def __init__(self, dataset):
super(HeterogeneousEmbeddingDataWrapper, self).__init__()
self.dataset = dataset
[docs] def train_wrapper(self):
return self.dataset.data
[docs] def test_wrapper(self):
return self.dataset.data