一种基于机器学习的视网膜反应预测系统

本发明公开一种基于机器学习的视网膜反应预测系统，包括：数据提取模块，用于从本地数据文件中提取视网膜电图（ERG）数据，根据ERG数据计算相关波形参数，并将相关波形参数进行整理，得到按照频率分类的ERG分析结果；特征工程模块，用于合并不同频率下的ERG分析结果，并在合并后的数据集中添加光强信息，以及对数据进行清理；模型训练模块，用于利用孤立森林算法检测并清除数据中的离群点，并通过使用不同的回归算法，根据数据分别训练得到正向预测模型和反向预测模型；特征预测模块，用于通过图形用户界面（GUI）提供交互操作，根据用户指令获取数据文件，并根据指令选择预训练的机器学习模型执行视网膜反应的正向或反向预测，提高了效率。