コード例 #1
0
ファイル: make_sv_lstm.py プロジェクト: kaneko-takayuki/NLP
    def convert(self, sentence, label):
        """
        文章からベクトルを生成する
        :param sentence: 英語文章
        :param label: ラベル
        :return: (入力ベクトルリスト, ラベルリスト)
        """
        # 入力ベクトルリストを取得
        inputs = vectorizer.sentence_vector(sentence)

        # labelを配列形式にして返す
        return inputs, [label]
コード例 #2
0
ファイル: e_w2v_svm.py プロジェクト: kaneko-takayuki/NLP
    def convert(self, sentence, label):
        """
        文章からベクトルを生成する
        :param sentence: 英語文章
        :param label: ラベル
        :return: (入力ベクトルリスト, ラベルリスト)
        """
        # 入力ベクトルリストを取得
        inputs = vectorizer.sentence_vector(sentence, self.window_size)

        # vectorsと同じ要素数のラベルリストを生成
        labels = [label for _ in range(len(inputs))]

        return inputs, labels