定义模型框架后使用经过预处理的数据集进行训练、测试模型,并将测试后的模型保存,以便调用。
模型训练相关代码如下:
模型保存的相关代码如下:
#模型保存 saver.save(sess,save_dir) print('Model Trained and Saved')