MLModel
MLModel是一个抽象类,它需要在具体的机器学习算法实现中进行继承。它是作为一个容器类,管理具体算法中的所有PSModel,作为一个整体模型被加载,训练和保存。同时,它需要负责Predict的执行逻辑实现。
核心方法
-
- 定义:
- 功能描述:根据模型和预测数据得到预测的结果,需要由具体的模型(PSModel)来实现预测流程
- 参数:storage: DataBlock[predictType] 预测数据
- 返回值:DataBlock[PredictResult] 预测结果
setSavePath
- 定义:
- 参数:conf: Configuration 任务配置信息
- 返回值:无
- setLoadPath
- 定义:
- 功能描述:设置模型的整体加载路径等
- 参数:conf: Configuration 任务配置信息
- 返回值:无
-
- 定义:
- 参数:
- name: String 模型名
- psModel: PSModel[_] 对应的PSModel对象
- 返回值:无
getPSModel
- 定义:
- 功能描述:根据矩阵名获取对应的PSModel对象
- 参数:name: String 矩阵名
- 返回值:PSModel对象