不同数据类型如何处理?
来源:4-4 tf.io.decode_csv使用
慕标9016513
2022-01-06
tf.io.decode_csv()之后通过tf.stack()获取x和y,但是tf.stack()要求所有字段的数据类型必须一样,如果有int、float、string等不同类型,如何处理?
写回答
1回答
-
正十七
2022-01-18
在我们的code中,x和y都是stack同一个字段的数据啊,同一个字段的数据都是同类型的。
def parse_csv_line(line, n_fields = 9): defs = [tf.constant(np.nan)] * n_fields parsed_fields = tf.io.decode_csv(line, record_defaults=defs) x = tf.stack(parsed_fields[0:-1]) y = tf.stack(parsed_fields[-1:]) return x, y
00
相似问题