Python LabelField.as_tensor 예제들

예제 #1

파일: label_field_test.py 프로젝트: ericyao2013/NLP-allenai_allennlp

    def test_label_field_can_index_with_vocab(self):
        vocab = Vocabulary()
        vocab.add_token_to_namespace("entailment", namespace="labels")
        vocab.add_token_to_namespace("contradiction", namespace="labels")
        vocab.add_token_to_namespace("neutral", namespace="labels")

        label = LabelField("entailment")
        label.index(vocab)
        tensor = label.as_tensor(label.get_padding_lengths()).data.cpu().numpy()
        numpy.testing.assert_array_almost_equal(tensor, numpy.array([0]))

예제 #2

파일: label_test.py 프로젝트: snowdj/explaining_predictions_with_LIME

    def test_label_field_can_index_with_vocab(self):
        vocab = Vocabulary()
        vocab.add_token_to_namespace("entailment", namespace="labels")
        vocab.add_token_to_namespace("contradiction", namespace="labels")
        vocab.add_token_to_namespace("neutral", namespace="labels")

        label = LabelField("entailment")
        label.index(vocab)
        tensor = label.as_tensor(label.get_padding_lengths())
        assert tensor.item() == 0

예제 #3

파일: label_field_test.py 프로젝트: apmoore1/allennlp

    def test_label_field_can_index_with_vocab(self):
        vocab = Vocabulary()
        vocab.add_token_to_namespace("entailment", namespace="labels")
        vocab.add_token_to_namespace("contradiction", namespace="labels")
        vocab.add_token_to_namespace("neutral", namespace="labels")

        label = LabelField("entailment")
        label.index(vocab)
        tensor = label.as_tensor(label.get_padding_lengths())
        assert tensor.item() == 0

예제 #4

파일: label_field_test.py 프로젝트: ericyao2013/NLP-allenai_allennlp

 def test_as_tensor_returns_integer_tensor(self):
     label = LabelField(5, skip_indexing=True)
     tensor = label.as_tensor(label.get_padding_lengths()).data.cpu().numpy()
     numpy.testing.assert_array_almost_equal(tensor, numpy.array([5]))

예제 #5

파일: label_test.py 프로젝트: snowdj/explaining_predictions_with_LIME

    def test_as_tensor_returns_integer_tensor(self):
        label = LabelField(5, skip_indexing=True)

        tensor = label.as_tensor(label.get_padding_lengths())
        assert tensor.item() == 5

예제 #6

파일: label_field_test.py 프로젝트: apmoore1/allennlp

    def test_as_tensor_returns_integer_tensor(self):
        label = LabelField(5, skip_indexing=True)

        tensor = label.as_tensor(label.get_padding_lengths())
        assert tensor.item() == 5