本期给大家介绍的是116用python来预测螃蟹的年龄,效果图如下:
这是调用训练好的模型来识别的,运行python 02pyqt.py的可视化界面,通过输入螃蟹的各项属性参数来预测最后螃蟹的年龄。
代码仓库或视频演示地址:
116用python来预测螃蟹的年龄_哔哩哔哩_bilibili
代码整体是非常简便的,总共两个py部分和一个数据集在data文件夹下。
Data数据如下,第一列是序号,第2至第9列是螃蟹的各项属性参数,最后一列是螃蟹的年龄。
运行01main.py中一共包含了9种机器学习模型,包含线性回归、随机森林、LightGBM、SVM、KNN、决策树、GBM、MLP、XGBoost等。训练结束后会弹出结果柱状图,其中LGBM的效果最佳,验证集上得分为0.6361。
最后运行02pyqt.py是个可视化的界面,加载训练好的LGBM模型,在界面上输入螃蟹的各项属性参数,最后螃蟹的品质即可。
Data.csv数据集可以替换自己的数据集来进行训练识别,方便简单替换数据集,只有保持n列去识别1列,即可实现。