本项目是一个基于LSTM的叉车臂速度预测模型,主要用于实现油泵电机的速度跟随和负载重量分类预测。
本项目使用的数据集是一个包含多个变量的时间序列数据集,其中包括油泵电机的实际速度、控制速度、期望速度与目标高度差。数据位于data文件夹中,train和test。
使用||符号进行分割,分别为: 文件名、本行数据的开始时间、结束时间、预测时间、预测速度(标签)、分类类别号(标签)、实际速度、控制速度、目标高度差。 robokit_2024-11-18_13-30-36.0.log.txt|s|133236.198|e|133237.148|l|133237.648 0.103 | 1.5 | 0.104 0.104 0.104 0.104 0.103 0.104 0.104 0.104 0.104 0.103 0.103 0.104 0.105 0.105 0.104 0.104 0.104 0.104 0.104 0.104 | 0.140 0.140 0.140 0.140 0.140 0.140 0.140 0.140 0.140 0.140 0.140 0.140 0.140 0.140 0.140 0.140 0.140 0.140 0.140 0.140 | 0.807 0.804 0.798 0.794 0.791 0.788 0.785 0.782 0.778 0.773 0.770 0.766 0.763 0.760 0.756 0.753 0.747 0.743 0.739 0.737 |
LSTM.py 训练和模型转换
modelLSTM.py LSTM模型定义
运行LSTM.py即可。