利用机器学习进行文本心理健康检测。

应用名称:BetterAid?

系统架构

这是一款基于移动应用程序的健康助手,集成了机器学习算法。通过该应用,用户可以通过OTP验证登录并创建个人资料。如果已登录,用户可以直接登录主页。登录后,用户将被引导至主页,在那里他们将拥有心理健康检测服务、BMI计算器以及医生信息面板。心理健康专家通过机器学习算法检测心理健康问题。为了实现机器学习算法,我们从头开始创建了一个数据集。对于数据集,我们通过网络抓取提取了社交媒体文本。“Instant Data Scraper”用于从Reddit提取数据。Reddit是丰富的文本质量数据集的绝佳来源。然后我们手动清理和注释了数据。Jupyter notebook被用作模型实现的源编辑器。我们在Notebook中将数据集合并为单个数据集。我们删除了停用词、进行了分词,并对数据集使用了词干提取技术。对于特征提取,使用了TF-IDF。使用了预训练的机器学习算法来训练模型。XGBoost和朴素贝叶斯被实现为机器学习模型。机器学习算法与Flask API集成,以便将机器学习算法集成到移动应用程序中。

登录和OTP面板

主页

医生信息

预测部分??

通过预测部分推荐医生

GitHub

查看 Github