def __init__(self, embedding: BaseEmbedding = None, hyper_parameters: Dict = None, **kwargs):
if embedding is None:
self.embedding = CustomEmbedding('custom', sequence_length=0, embedding_size=100)
else:
self.embedding = embedding
self.model: Model = None
self._hyper_parameters_ = self.__base_hyper_parameters__.copy()
self._label2idx = {}
self._idx2label = {}
self.model_info = {}
self.task = 'classification'
if hyper_parameters:
self._hyper_parameters_.update(hyper_parameters)
def setUpClass(cls):
cls.embedding = CustomEmbedding('empty_embedding',
sequence_length=SEQUENCE_LENGTH,
embedding_size=100)
corpus = [['我', '们', '变', '而', '以', '书', '会', '友', '，', '以', '书', '结', '缘', '，',
'把', '欧', '美', '、', '港', '台', '流', '行', '的',
'食', '品', '类', '图', '谱', '、', '画', '册', '、',
'工', '具', '书', '汇', '集', '一', '堂', '。'],
['为', '了', '跟', '踪', '国', '际', '最', '新', '食', '品',
'工', '艺', '、', '流', '行', '趋', '势', '，', '大', '量',
'搜', '集', '海', '外', '专', '业', '书', '刊', '资', '料',
'是', '提', '高', '技', '艺', '的', '捷', '径', '。'],
['其', '中', '线', '装', '古', '籍', '逾', '千', '册',
'；', '民', '国', '出', '版', '物', '几', '百', '种',
'；', '珍', '本', '四', '册', '、', '稀', '见', '本',
'四', '百', '余', '册', '，', '出', '版', '时', '间',
'跨', '越', '三', '百', '余', '年', '。'],
['有', '的', '古', '木', '交', '柯', '，',
'春', '机', '荣', '欣', '，', '从', '诗',
'人', '句', '中', '得', '之', '，', '而',
'入', '画', '中', '，', '观', '之', '令', '人', '心', '驰', '。', '我']]
cls.embedding.build_token2idx_dict(x_data=corpus, min_count=2)
def __init__(self, *args, **kwargs):
super(CustomEmbeddingsTest, self).__init__(*args, **kwargs)
self.embedding = CustomEmbedding('empty_embedding',
sequence_length=SEQUENCE_LENGTH,
embedding_size=100)