116用python来预测螃蟹的年龄

本期给大家介绍的是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列，即可实现。