Commit 7389f013 authored by 赵威's avatar 赵威

update type

parent 989a035d
......@@ -182,7 +182,7 @@ def model_predict2(device_id, diary_ids, device_dict, diary_dict, predict_fn):
for col in float_columns:
features[col] = _float_feature(float(tmp[col]))
for col in str_columns:
features[col] = _bytes_feature(tmp[col])
features[col] = _bytes_feature(str(tmp[col]).encode(encoding="utf-8"))
example = tf.train.Example(features=tf.train.Features(feature=features))
examples.append(example.SerializeToString())
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment