MLModel

---

MLModel表示一个机器学习算法中，需要用到的完整模型集合。一个复杂的机器学习算法，往往需要多个远程模型PSModel协作，为了方便对这些模型进行统一操作和管理，MLModel作为一个抽象类，成为了Angel所有模型的基类

功能

MLModel是一个抽象类，它需要在具体的机器学习算法实现中进行继承。它是作为一个容器类，管理具体算法中的所有PSModel，作为一个整体模型被加载，训练和保存。同时，它需要负责Predict的执行逻辑实现。

核心方法

• predict

  • 定义：def predict(storage: DataBlock[predictType]): DataBlock[PredictResult]
  • 功能描述：根据模型和预测数据得到预测的结果，需要由具体的模型（PSModel）来实现预测流程
  • 参数：storage: DataBlock[predictType] 预测数据
  • 返回值：DataBlock[PredictResult] 预测结果

• setSavePath

  • 定义：def setSavePath(conf: Configuration)
  • 功能描述：设置需要保存的模型，以及保存的路径等
  • 参数：conf: Configuration 任务配置信息
  • 返回值：无

• setLoadPath

  • 定义：def setLoadPath(conf: Configuration)
  • 功能描述：设置模型的整体加载路径等
  • 参数：conf: Configuration 任务配置信息
  • 返回值：无

• addPSModel

  • 定义：def addPSModel(name: String, psModel: PSModel[_])
  • 功能描述：在模型中新增一个PSModel
  • 参数：
    • name: String 模型名
    • psModel: PSModel[_] 对应的PSModel对象
  • 返回值：无

• getPSModel

  • 定义：def getPSModel(name: String): PSModel[_]
  • 功能描述：根据矩阵名获取对应的PSModel对象
  • 参数：name: String 矩阵名
  • 返回值：PSModel对象