为什么我不能使用tfdv.infer_schema()生成模式？

Question

"TypeError:统计是StatsOptions类型的，应该是DatasetFeatureStatisticsList proto。“错误显示当我使用tfdv.infer_schema()选项生成模式时，但是当我使用tfdv.StatsOptions类使用feature_allowlist筛选相关特性时，我无法这样做。有人能帮我吗？

features_remove= {"region","fiscal_week"}

columns= [col for col in df.columns if col not in features_remove]
stat_Options= tfdv.StatsOptions(feature_allowlist=columns)
print(stat_Options.feature_allowlist)


schema= tfdv.infer_schema(stat_Options)

---------------------------------------------------------------------------
TypeError                                 Traceback (most recent call last)
<ipython-input-53-e61b2454028e> in <module>
----> 1 schema= tfdv.infer_schema(stat_Options)
      2 schema

C:\ProgramData\Anaconda3\lib\site-packages\tensorflow_data_validation\api\validation_api.py in infer_schema(statistics, infer_feature_shape, max_string_domain_size, schema_transformations)
     95   """
     96   if not isinstance(statistics, statistics_pb2.DatasetFeatureStatisticsList):
---> 97     raise TypeError(
     98         'statistics is of type %s, should be '
     99         'a DatasetFeatureStatisticsList proto.' % type(statistics).__name__)

TypeError: statistics is of type StatsOptions, should be a DatasetFeatureStatisticsList proto.

Answer 1

原因很简单，您必须将一个statistics_pb2.DatasetFeatureStatisticsList对象传递给tfdv.infer_schema函数，而不是statsOptions。

你应该往这边走：

features_remove= {"region","fiscal_week"}

columns= [col for col in df.columns if col not in features_remove]
stat_Options= tfdv.StatsOptions(feature_allowlist=columns)
print(stat_Options.feature_allowlist)

stats = tfdv.generate_statistics_from_dataframe(df, stat_Options)
schema= tfdv.infer_schema(stats)