当前位置: 首页>>代码示例>>Python>>正文


Python Dataset.get_tokens方法代码示例

本文整理汇总了Python中dataset.Dataset.get_tokens方法的典型用法代码示例。如果您正苦于以下问题:Python Dataset.get_tokens方法的具体用法?Python Dataset.get_tokens怎么用?Python Dataset.get_tokens使用的例子?那么恭喜您, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在dataset.Dataset的用法示例。


在下文中一共展示了Dataset.get_tokens方法的3个代码示例,这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞,您的评价将有助于系统推荐出更棒的Python代码示例。

示例1: train

# 需要导入模块: from dataset import Dataset [as 别名]
# 或者: from dataset.Dataset import get_tokens [as 别名]
    def train(self, session, dataset: Dataset):
        word_vectors = []
        for tokens in dataset.get_tokens():
            word_vectors.append(self.encode_word_vector(tokens))

        slot_vectors = []
        for iob in dataset.get_iob():
            slot_vectors.append(self.encode_slot_vector(iob))

        cost_output = float('inf')
        for _ in range(self.__step_per_checkpoints):
            indexes = np.random.choice(len(word_vectors), self.__batch_size, replace=False)
            x = [word_vectors[index] for index in indexes]
            y = [slot_vectors[index] for index in indexes]

            self.__step += 1
            _, cost_output = session.run([self.__optimizer, self.__cost],
                                         feed_dict={self.__x: x,
                                                    self.__y: y,
                                                    self.__dropout: 0.5})

        checkpoint_path = os.path.join('./model', "slot_filling_model.ckpt")
        self.__saver.save(session, checkpoint_path, global_step=self.__step)

        return cost_output
开发者ID:fin10,项目名称:MachineLearningStudy,代码行数:27,代码来源:slot_tagger_model.py

示例2: test_append

# 需要导入模块: from dataset import Dataset [as 别名]
# 或者: from dataset.Dataset import get_tokens [as 别名]
 def test_append(self):
     domain = 'test'
     raw = 'i wanna go to (Essex Street)[to_stop]'
     iob = ['o', 'o', 'o', 'o', 'b-test.to_stop', 'i-test.to_stop']
     tokens = ['i', 'wanna', 'go', 'to', 'essex', 'street']
     dataset = Dataset()
     dataset.append(domain, raw)
     self.assertEqual(dataset.get_domain(0), domain)
     self.assertEqual(dataset.get_raw(0), raw)
     self.assertEqual(dataset.get_iob(0), iob)
     self.assertListEqual(dataset.get_tokens(0), tokens)
开发者ID:fin10,项目名称:MachineLearningStudy,代码行数:13,代码来源:dataset_test.py

示例3: test

# 需要导入模块: from dataset import Dataset [as 别名]
# 或者: from dataset.Dataset import get_tokens [as 别名]
    def test(self, session, dataset: Dataset):
        word_vectors = []
        for tokens in dataset.get_tokens():
            word_vectors.append(self.encode_word_vector(tokens))

        slot_vectors = []
        for iob in dataset.get_iob():
            slot_vectors.append(self.encode_slot_vector(iob))

        prediction_output, score_output, length_output = session.run(
            [self.__prediction, self.__score, self.__length],
            feed_dict={self.__x: word_vectors,
                       self.__y: slot_vectors,
                       self.__dropout: 1.0})

        predict = []
        for i in range(dataset.length()):
            predict.append(self.decode_slot_vector(prediction_output[i][:length_output[i]]))

        return score_output, predict
开发者ID:fin10,项目名称:MachineLearningStudy,代码行数:22,代码来源:slot_tagger_model.py


注:本文中的dataset.Dataset.get_tokens方法示例由纯净天空整理自Github/MSDocs等开源代码及文档管理平台,相关代码片段筛选自各路编程大神贡献的开源项目,源码版权归原作者所有,传播和使用请参考对应项目的License;未经允许,请勿转载。