【最佳实践】使用Spark MLlib实现车辆满意度测评
使用KNN分类算法进行车辆满意度测评,选用经典的Car Evaluation数据集,通过对六个特征值进行测评,得到用户对车辆的满意程度。
KNN Classification(K最近邻分类)算法用于产生二分类或多分类模型。基本思想是,如果一个样本在特征空间k个最邻近样本中,大多数属于某一个类别,则该样本也属于这个类别。该算法对噪声具有较好的鲁棒性,并且可以较好的避免样本数量不平衡的问题。其中k是用户指定的整数值,该参数的选取高度依赖输入数据,通常较大的k会抑制噪声的影响,但同时也会使得分类边界变得模糊。该算法适合数据记录数小于100万,维度小于100维的分类场景。在二分类问题中,选取k为奇数有助于避免两个分类平票的情形。
本示例介绍如何使用Spark MLlib引擎的KNN分类算法实现车辆满意度测评应用。车辆满意度测评样例的操作流程如下所示。
- 准备数据:下载数据集、示例代码,然后上传至OBS桶中。
- 训练模型:编写基于Spark MLlib中ALS算法的模型训练脚本,新建训练作业进行模型训练。
- 部署模型:得到训练好的模型文件后,新建预测作业将模型部署为在线预测服务。
- 预测结果:发起预测请求获取预测结果。
准备数据
ModelArts提供了用于训练的数据集和示例代码,执行如下步骤,下载数据集和示例代码,并上传至OBS中。
- 在gitee的ModelArts-Lab工程中,单击“克隆/下载”,然后单击“下载ZIP”,下载工程。
- 下载完成后,解压缩“ModelArts-Lab-master.zip”文件,然后在“\ModelArts-Lab-master\official_examples\Using_Spark_MLlib_to_Create_a_Car_Evaluation_Application”目录中获取到训练数据集和示例代码。
表1 文件说明 文件名称
说明
“car.csv”
训练数据集。数据集的详情如表2所示。
“car_meta.desc”
元数据文件。
“knn_classification.py”
使用ALS算法编写的训练脚本。
“customize_service.py”
用户自定义的预测脚本,需与训练脚本“knn_classification.py”存储在相同OBS目录下。训练过程中,将被自动拷贝至相应的模型目录下。
- 进入OBS管理控制台,新建桶和文件夹,分别用于存储训练数据集和示例代码。例如新建“test-modelarts2”桶,并在此桶下新建“sparkml/car/data”和“sparkml/car/code”文件夹。
- 将步骤2中获取的文件,上传至对应OBS路径下,即“sparkml/car/data”和“sparkml/car/code”文件夹。OBS上传文件的操作指导,请参见上传文件。
训练模型
- 在ModelArts管理控制台,进入“训练管理 > 训练作业”页面,单击左上角的“创建”。
- 如图2 创建训练作业-基本信息和图3 创建训练作业-详细参数 所示,参考图中示例,填写训练作业相关参数,然后单击“下一步”。
其中,“数据来源”和“算法来源”即准备数据上传的OBS路径及文件。“训练输出位置”,建议新建一个OBS文件夹,用于存储训练输出的模型及其预测文件,例如“test-modelarts/car/output”。
- 在规格确认页面,确认信息无误后,单击“提交”。
- 在“训练作业”管理页面,当训练作业变为“运行成功”时,即完成了模型训练过程。如有问题,可单击作业名称,进入作业详情界面查看训练作业日志信息。
训练作业需要花费一些时间,预计十几分钟。当训练时间超过一定时间(如1个小时),请及时手动停止,释放资源。否则会导致欠费,尤其对于使用GPU训练的模型项目。
预测结果
待部署模型运行完成后,可以验证发布的预测服务是否正常。
- 在“部署上线>在线服务”页面,单击服务名称进入详情页面
- 在“预测”页签,参考图6 测试服务所示样例,输入预测代码,然后单击“预测”。在右侧“返回结果”中,查看预测结果。
预测请求示例代码如下所示。
{ "data": { "req_data": [ { "buying_price": "high", "maint_price": "high", "doors": "2", "persons": "2", "lug_boot": "small", "safety": "low", "acceptability": "acc" }, { "buying_price": "high", "maint_price": "high", "doors": "2", "persons": "2", "lug_boot": "small", "safety": "low", "acceptability": "acc" } ] } }
- 在“调用指南”页签,可以获取调用API接口,并使用Postman工具进行测试。
- 点赞
- 收藏
- 关注作者
评论(0)