python - Tensorflow GradientBoostedDecisionTreeClassifier 错误：“密集浮点特征必须是矩阵”

Answer 1

我猜测 tf.transform 创建的解析规范与我们通常得到的不同。你能分享一下transformed_metadata.schema.as_feature_spec()的输出吗？

作为一种解决方法，请尝试在 features = tf.train.shuffle_batch(...) 之后将此行添加到您的 input_fn：

features = {feature_name: tf.reshape(feature_value, [-1, 1]) for feature_name, feature_value in features.items()}