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
    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]))
Пример #3
0
    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
0
    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)
        array = label.pad(label.get_padding_lengths())
        numpy.testing.assert_array_almost_equal(array[0],
                                                numpy.array([1, 0, 0]))
 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]))
    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
Пример #7
0
 def test_as_array_returns_integer_array(self):
     label = LabelField(5, skip_indexing=True)
     array = label.as_array(label.get_padding_lengths())
     numpy.testing.assert_array_almost_equal(array, numpy.array([5]))
Пример #8
0
 def test_pad_returns_one_hot_array(self):
     label = LabelField(5, num_labels=10)
     array = label.as_array(label.get_padding_lengths())
     numpy.testing.assert_array_almost_equal(
         array, numpy.array([0, 0, 0, 0, 0, 1, 0, 0, 0, 0]))
Пример #9
0
    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