在 TFX 中,是否可以根据日期推断架构?

In TFX, is it possible to infer Schema with dates?

我将 TFX(更准确地说是 TensorFlow 数据验证)与 infer_schema 中记录的方法一起使用 https://www.tensorflow.org/tfx/data_validation/api_docs/python/tfdv/infer_schema。它从描述列类型的 csv 文件生成模式。

它适用于浮点数、字节数、类别...但我还想检测日期。我没有在教程或指南中找到它。生成的原型消息支持日期,因此这不是问题(请参阅 TimeDomain)。 https://github.com/tensorflow/metadata/blob/master/tensorflow_metadata/proto/v0/schema.proto

我尝试使用该格式(非美国日期格式)的 CSV 文件,它被识别为字节 :(

date, amount
15/08/2001, 0.3120682494
16/08/2001, 0.9310268917
17/08/2001, 0.902986235

代码和教程里的一样,所以差不多:

train_stats = tfdv.generate_statistics_from_csv(data_location="/content/csv_with_dates.csv")
schema = tfdv.infer_schema(statistics=train_stats)
tfdv.display_schema(schema=schema)

其中显示:

Type    Presence    Valency Domain
Feature name                
'date'  BYTES   required        -
'amount'    FLOAT   required        -

我可以让它工作吗?怎么样?

谢谢!

目前没有,可能在即将推出的版本中。 如果您检查您提到的 link,您会发现功能支持以下类型(不包括日期):

enum FeatureType {
  TYPE_UNKNOWN = 0;
  BYTES = 1;
  INT = 2;
  FLOAT = 3;
  STRUCT = 4;
}