3703673d516335d7c5188b954e6bf7d123c646ba,tensorflow_transform/beam/impl_test.py,BeamImplTest,testStringToTFIDF,#BeamImplTest#,809

Before Change

    expected_transformed_data = [{
        "tf_idf": [0, 0, 0]
    }, {
        "tf_idf": [0, (1/4)*log_3_over_2, 0, 0]
    }, {
        "tf_idf": [(1/5)*log_3_over_2, (1/5)*log_3_over_2, (1/5)*log_3_over_2,
                   (1/5)*log_3_over_2, (1/5)*log_3_over_2]
    }]

After Change

    expected_transformed_schema = dataset_metadata.DatasetMetadata({
        "tf_idf": sch.ColumnSchema(tf.float32, [None],
                                   sch.ListColumnRepresentation()),
        "index": sch.ColumnSchema(tf.int64, [None],
                                  sch.ListColumnRepresentation())
    })
    self.assertAnalyzeAndTransformResults(
        input_data, input_schema, preprocessing_fn, expected_transformed_data,
        expected_transformed_schema)

Instances