提交 5f4792fa 编写于 作者: H Hongyi 提交者: Jiangtao Hu

Prediction: small lint fix

上级 9630fa6e
......@@ -29,7 +29,7 @@ import tensorflow as tf
import proto.fnn_model_pb2
from proto.fnn_model_pb2 import FnnModel, Layer
dim_input = 3+60
dim_input = 3 + 60
dim_output = 12
def load_data(filename):
......@@ -106,4 +106,4 @@ if __name__ == "__main__":
model.fit(X_train, Y_train, epochs=10)
model_path = os.path.join(os.getcwd(), "junction_mlp_model.bin")
save_model(model, model_path)
print("Model saved to: " + model_path)
\ No newline at end of file
print("Model saved to: " + model_path)
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册