尧图网站设计 尧图网站设计YAOTU DESIGN
ARTICLE DETAIL

资讯详情

深耕网站设计与一线实操的经验洞察。

PredictionIO 算法切换实战:将 Classification 模板从 Naive Bayes 替换为 MLlib Random Forests

PredictionIO 算法切换实战:将 Classification 模板从 Naive Bayes 替换为 MLlib Random Forests 机器学习后端大数据【免费下载链接】predictionioPredictionIO, a machine learning server for developers and ML engineers.项目地址https://gitcode.com/gh_mirrors/pr/predictionio点击查看免费下载PredictionIO 的每个引擎模板Engine Template都内置了默认算法而引擎可以在 Algorithm 类中调用几乎任意算法。本文以 Classification分类引擎模板为例完整演示如何在不改动数据流的前提下将默认的朴素贝叶斯Naive Bayes算法切换为 Apache Spark MLlib 的随机森林Random Forests算法覆盖新算法类的编写、引擎工厂注册、engine.json 参数配置以及构建、训练、部署的完整链路。读完本文你将掌握 PredictionIO 引擎算法替换的标准流程并能在自己的引擎中自由切换、组合多种算法。算法切换的核心机制一切从 Algorithm 类开始PredictionIO 的引擎架构遵循 DASE 模式DataSource、Algorithm、Serving、Evaluator。算法位于整条数据流水线的中间环节它接收PreparedData作为训练输入产出可序列化的模型并在预测阶段把Query映射为PredictedResult。在 docs/manual/source/algorithm/index.html.md 中明确指出引擎可以在 Algorithm 类中调用几乎任意算法PredictionIO 目前对 Apache Spark MLlib 机器学习库提供原生支持模板库template gallery中的多个引擎模板都基于它构建。因此切换算法本质上只做一件事——修改 Algorithm 类。引擎模板的其他部分DataSource、Preparator、Serving可以完全不动前提是新旧算法消费相同的训练数据结构。实战场景Classification 模板切换到 MLlib Random ForestsClassification 引擎模板默认使用 MLlib 的 Naive Bayes 算法本实战将把默认算法替换为 MLlib Random Forests 算法。完整修改后的源码可直接在本仓库的 examples/scala-parallel-classification/add-algorithm 中找到下面的每一步都可与该目录下的文件一一对照。前置准备定位引擎目录与算法文件如果你已按照 Classification 引擎模板快速入门 创建了名为MyClassification的引擎那么算法文件位于MyClassification/src/main/scala/NaiveBayesAlgorithm.scala复制该文件并新建RandomForestAlgorithm.scala接下来的所有改动都在这个新文件中进行。第一步定义新算法类与参数1. 更换 import将 Naive Bayes 的 import 替换为 MLlib Random Forests 所需的类// 原代码 import org.apache.spark.mllib.classification.NaiveBayes import org.apache.spark.mllib.classification.NaiveBayesModel // 改为 import org.apache.spark.mllib.tree.RandomForest // CHANGED import org.apache.spark.mllib.tree.model.RandomForestModel // CHANGEDRandomForest是训练入口RandomForestModel是训练产出的模型对象。2. 定义 RandomForestAlgorithmParams 参数类// CHANGED case class RandomForestAlgorithmParams( numClasses: Int, numTrees: Int, featureSubsetStrategy: String, impurity: String, maxDepth: Int, maxBins: Int ) extends Params该参数类对应 MLlib Random Forest 分类器的核心超参数各参数含义如下参数类型说明numClassesInt分类类别数量即标签的取值个数numTreesInt随机森林中决策树的数量树越多模型通常越稳健但训练开销越大featureSubsetStrategyString每棵树分裂时考虑的特征子集策略如auto、sqrt等impurityInt不纯度度量方式分类任务常用gini也可用entropymaxDepthInt每棵树的最大深度控制模型复杂度防止过拟合maxBinsInt连续特征离散化时使用的最大分箱数这些参数的具体取值会在后面通过engine.json注入这正是 PredictionIO 参数化算法的标准做法算法参数由配置文件驱动改算法不需要改代码。3. 定义 RandomForestAlgorithm 类// extends P2LAlgorithm because the MLlibs RandomForestModel doesnt // contain RDD. class RandomForestAlgorithm(val ap: RandomForestAlgorithmParams) // CHANGED extends P2LAlgorithm[PreparedData, RandomForestModel, // CHANGED Query, PredictedResult] { // CHANGED def train(sc: SparkContext, data: PreparedData): RandomForestModel { // CHANGED // Empty categoricalFeaturesInfo indicates all features are continuous. val categoricalFeaturesInfo Map[Int, Int]() RandomForest.trainClassifier( data.labeledPoints, ap.numClasses, categoricalFeaturesInfo, ap.numTrees, ap.featureSubsetStrategy, ap.impurity, ap.maxDepth, ap.maxBins) } def predict( model: RandomForestModel, // CHANGED query: Query): PredictedResult { val label model.predict(Vectors.dense( Array(query.attr0, query.attr1, query.attr2) )) PredictedResult(label) } }与原 NaiveBayes 算法相比改动要点如下泛型中的模型类型由NaiveBayesModel改为RandomForestModeltrain()返回RandomForestModel内部调用RandomForest.trainClassifier并通过空的categoricalFeaturesInfoMap[Int, Int]()告知 MLlib 所有特征都是连续型特征predict()接收RandomForestModel将查询中的attr0、attr1、attr2组装成稠密向量后调用model.predict得到预测标签算法仍然继承P2LAlgorithm因为 MLlib 的RandomForestModel内部不包含 RDD训练产出的是局部模型而非分布式的PModel。仓库中该文件的完整实现见 RandomForestAlgorithm.scala原朴素贝叶斯实现见 NaiveBayesAlgorithm.scala两者对比可以清晰地看出算法替换的差异面。为什么 DataSource 和 PreparedData 不用改因为 MLlib 的 Random Forest 与 Naive Bayes 消费的是同一种训练数据——RDD[LabeledPoint]。本模板的 DataSource 通过PEventStore.aggregateProperties从事件存储中聚合用户属性plan、attr0、attr1、attr2再映射为LabeledPoint见 DataSource.scala这条链路对新旧算法完全透明。如果新算法需要不同类型的训练数据则必须同步修改 DataSource 和 PreparedData 来适配。第二步在 Engine.scala 中注册新算法打开Engine.scala修改EngineFactory把新定义的RandomForestAlgorithm加入算法映射表并命名为randomforestobject ClassificationEngine extends IEngineFactory { def apply() { new Engine( classOf[DataSource], classOf[Preparator], Map(naive - classOf[NaiveBayesAlgorithm], randomforest - classOf[RandomForestAlgorithm]), // ADDED classOf[Serving]) } }这个 Map 是 PredictionIO 算法注册的核心key 是算法在配置文件中的引用名value 是对应的算法类。现在引擎同时拥有两个已注册算法——naive和randomforest后续通过engine.json指定实际使用哪一个。仓库中对应实现见 Engine.scala。第三步更新 engine.json 指定算法与参数engine.json是引擎的运行时配置。将algorithms数组中的name改为randomforest并填入随机森林的超参数... algorithms: [ { name: randomforest, params: { numClasses: 4, numTrees: 5, featureSubsetStrategy: auto, impurity: gini, maxDepth: 4, maxBins: 100 } } ] ...这里的params会反序列化为RandomForestAlgorithmParams实例并注入算法构造函数与代码中的参数类一一对应。仓库中的实际配置文件 engine.json 还包含id、description、engineFactory指向org.apache.predictionio.examples.classification.ClassificationEngine以及datasourceappName: MyApp1等顶层配置项可一并参考。第四步构建、训练与部署完成上述三处修改后按标准流程重新构建、训练并部署引擎$ pio build $ pio train $ pio deploy引擎即开始使用MLlib Random Forests算法提供预测服务而非默认的 Naive Bayes 算法。查询方式与模板保持一致例如向部署引擎发送{ attr0:2, attr1:0, attr2:0 }即可获得预测的plan标签。切回默认算法只需一步由于两个算法都已注册在EngineFactory中把engine.json中algorithms[0].name改回naive并将params换成 Naive Bayes 的参数如{ lambda: 1.0 }重新pio build pio train pio deploy即可无需改动任何代码——这正是算法与配置分离设计带来的灵活性。从源码看算法切换的完整数据流结合仓库中的完整示例工程 examples/scala-parallel-classification/add-algorithm可以看到算法切换后整个 DASE 流水线的实际运转方式DataSourceDataSource.scala从事件存储按appName聚合user实体的plan、attr0、attr1、attr2属性映射为RDD[LabeledPoint]标签plan 特征向量[attr0, attr1, attr2]其readEval方法还支持按evalK做 K 折切分供评估流程使用Preparator将TrainingData转换为PreparedData此处数据结构不变Algorithm由randomforest指向的RandomForestAlgorithm训练出RandomForestModel并实现Query→PredictedResult的预测映射Serving封装预测结果对外提供服务。从源码结构看该示例工程还包含 Evaluation.scala、CompleteEvaluation.scala 与 PrecisionEvaluation.scala说明切换后的随机森林算法同样可以接入模板的离线评估与参数调优流程用于比较不同算法的精确率等指标。扩展多算法并存与更多模板算法切换机制不限于 Classification 模板它是所有 PredictionIO 引擎的通用能力多算法并存如 Engine.scala 所示EngineFactory中的 Map 可以同时注册多个算法配合engine.json即可在多个算法间随时切换甚至为 A/B 对比评估提供基础SimilarProduct 模板的多算法多事件示例仓库中的 scala-parallel-similarproduct/multi-events-multi-algos 展示了单个引擎内组合多个算法与多种事件的更复杂用法并提供了engine.json与engine-cooccurrence.json两套配置其他模板的算法定制Classification 模板的完整定制教程含 DASE 组件详解见 docs/manual/source/templates/classification/dase.html.md.erb推荐阅读 scala-parallel-classification 目录下的多个变体示例如 reading-custom-properties加深理解。小结切换 PredictionIO 引擎算法的完整套路可以概括为四步写新算法类含参数类与 train/predict 逻辑→ 在 EngineFactory 中注册 → 在 engine.json 中指定算法名与参数 → pio build / train / deploy。只要新旧算法消费相同结构的训练数据数据层完全不需要改动即使数据结构不同也只需扩展 DataSource 与 PreparedData 即可。理解这一机制后你便可以在自己的引擎中自由接入 MLlib 的各类算法并通过配置文件随时切换与组合。赞分享机器学习后端大数据【免费下载链接】predictionioPredictionIO, a machine learning server for developers and ML engineers.项目地址https://gitcode.com/gh_mirrors/pr/predictionio点击查看免费下载相关推荐Apache Spark MLlib 集成学习实战Random Forests 与 Gradient-Boosted TreesRDD-based API深度指南Apache Spark MLlib 集成学习实战Random Forests 与 Gradient Boosted TreesRDD based API大数据数据分析批处理流处理机器学习图计算Hugo 模板中的 Duration.Seconds 方法将 time.Duration 精确换算为秒数Hugo 模板中的 Duration.Seconds 方法将 time.Duration 精确换算为秒数 Hugo 模板系统提供了一套基于 Go 标准库 ti开发工具前端CLIPredictionIO 内置算法库解析基于 Spark MLlib 的 Algorithm 体系与多算法引擎实战PredictionIO 内置算法库解析基于 Spark MLlib 的 Algorithm 体系与多算法引擎实战 Apache PredictionIO 的机器学习后端推荐系统创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考
返回列表