运行张量流时出现错误消息。
example.features.feature [col_name] .bytes_list.value.append(val.strip()) typeError:“ Self-emp-not-inc”的类型为str,但应为以下字节之一
numerical_features = ['age','fnlwgt','education-num','capital-gain', “资本损失”,“每周工时”]
df = pandas.read_csv(DATA_FILE).values 使用tf.python_io.TFRecordWriter(OUTPUT_FILE)作为作者: 对于df中的行: 示例= tf.train.Example() 对于col_name,zip中的val(column_names,行): 如果col_name在numeric_features中: example.features.feature [col_name] .int64_list.value.append(val) 其他: example.features.feature [col_name] .bytes_list.value.append(val.strip()) writer.write(example.SerializeToString())
它应该成功处理。