预测

# 预测 Model_Predict.v1

(可视化模块)
定义

用于深度学习模型的预测

M.dl_model_predict.v1(trained_model=None,input_data=None,batch_size=None, n_gpus=0)

参数

  • trained_model(DataSDource)—训练模型,来自Model_Train模型的输出
  • input_data(DataSDource)—验证集数据
  • batch_size—进行梯度下降时每个batch包含的样本数据集大小,训练时一个batch会被计算一次梯度下降使目标函数优化一步。
  • n_gpus—GPU数量,
  • verbose—日志输出规则:'0: 不显示';'1: 输出进度条记录';'2: 每个epoch输出一行记录'

返回

深度学习模型在验证集数据上的预测值

返回类型:

Outputs