使用朴素贝叶斯模型进行预测

Question

大家好，我是 scala 和 spark MLIB 的新手。

我有一个疾病数据集以及以下格式的症状：

疾病，症状1 症状2 症状3

我在 CSV 文件中有近 300 个上述格式的条目。

我想实现以下功能：

如果用户输入了症状，即症状 1、症状 2、症状 3，则模型必须能够预测疾病。

我有以下问题：

1. 我应该使用哪种机器学习模型来实现此功能。

2. 我已经研究了一些模型，如果有错误请纠正我。

3. 我可以向朴素贝叶斯模型提供文本输入吗？

4. 是否有任何示例代码可用于实现此功能。

Answer 1

您可以使用 Spark MLlib 中存在的任何分类算法作为进一步参考，请阅读官方文档并通过 databricks 博客https://databricks.com/blog/2015/07/29/new-features-in-machine-learning-pipelines-in-spark-1-4.html 中的此链接访问