plt.plot(model.predict(test_X).reshape(-1),label='pre') plt.plot(test_Y, color='#800080',label='true') plt.legend() plt.show()
时间: 2024-01-18 22:02:51 浏览: 41
这段代码看起来像是在使用 matplotlib 绘制预测数据和真实数据的对比图。其中,model.predict(test_X) 是用训练好的模型对测试数据进行预测,reshape(-1) 是将预测结果转为一维数组,test_Y 是测试数据的真实标签。plt.plot 是绘制折线图,plt.legend 是添加图例信息,plt.show 是展示图像。需要注意的是,这段代码缺少一些必要的库和数据,需要在上下文中补充完整才能运行。
相关问题
import pandas as pd import numpy as np import matplotlib.pyplot as plt import tensorflow as tf from tensorflow.keras.models import Sequential from tensorflow.keras.layers import LSTM, Dense data = pd.read_csv('车辆:274序:4结果数据.csv') x = data[['车头间距', '原车道前车速度']].values y = data['本车速度'].values train_size = int(len(x) * 0.7) test_size = len(x) - train_size x_train, x_test = x[0:train_size,:], x[train_size:len(x),:] y_train, y_test = y[0:train_size], y[train_size:len(y)] from sklearn.preprocessing import MinMaxScaler scaler = MinMaxScaler(feature_range=(0, 1)) x_train = scaler.fit_transform(x_train) x_test = scaler.transform(x_test) model = Sequential() model.add(LSTM(50, input_shape=(2, 1))) model.add(Dense(1)) model.compile(loss='mean_squared_error', optimizer='adam') history = model.fit(x_train.reshape(-1, 2, 1), y_train, epochs=100, batch_size=32, validation_data=(x_test.reshape(-1, 2, 1), y_test)) plt.plot(history.history['loss']) plt.plot(history.history['val_loss']) plt.title('Model loss') plt.ylabel('Loss') plt.xlabel('Epoch') plt.legend(['Train', 'Test'], loc='upper right') plt.show() train_predict = model.predict(x_train.reshape(-1, 2, 1)) test_predict = model.predict(x_test.reshape(-1, 2, 1)) train_predict = scaler.inverse_transform(train_predict) train_predict = train_predict.reshape(-1) # 将结果变为一维数组 y_train = scaler.inverse_transform(y_train.reshape(-1, 1)).reshape(-1) # 将结果变为一维数组 test_predict = scaler.inverse_transform(test_predict) y_test = scaler.inverse_transform([y_test]) plt.plot(y_train[0], label='train') plt.plot(train_predict[:,0], label='train predict') plt.plot(y_test[0], label='test') plt.plot(test_predict[:,0], label='test predict') plt.legend() plt.show()报错Traceback (most recent call last): File "C:\Users\马斌\Desktop\NGSIM_data_processing\80s\lstmtest.py", line 42, in <module> train_predict = scaler.inverse_transform(train_predict) File "D:\python\python3.9.5\pythonProject\venv\lib\site-packages\sklearn\preprocessing\_data.py", line 541, in inverse_transform X -= self.min_ ValueError: non-broadcastable output operand with shape (611,1) doesn't match the broadcast shape (611,2)
这个错误发生在将归一化后的预测结果进行逆归一化时。根据错误提示,训练集的预测结果和真实值的形状为 (611,),而测试集的预测结果和真实值的形状为 (1, 611)。这是因为在进行逆归一化时,要求预测结果和真实值的形状必须一致,即都为一维数组。
你可以将 y_test 的形状改为 (611,),方法是将 y_test 的数据类型从二维数组转换为一维数组,就像对 y_train 做的那样。修改后的代码如下:
```python
import pandas as pd
import numpy as np
import matplotlib.pyplot as plt
import tensorflow as tf
from tensorflow.keras.models import Sequential
from tensorflow.keras.layers import LSTM, Dense
data = pd.read_csv('车辆:274序:4结果数据.csv')
x = data[['车头间距', '原车道前车速度']].values
y = data['本车速度'].values
train_size = int(len(x) * 0.7)
test_size = len(x) - train_size
x_train, x_test = x[0:train_size,:], x[train_size:len(x),:]
y_train, y_test = y[0:train_size], y[train_size:len(y)]
from sklearn.preprocessing import MinMaxScaler
scaler = MinMaxScaler(feature_range=(0, 1))
x_train = scaler.fit_transform(x_train)
x_test = scaler.transform(x_test)
model = Sequential()
model.add(LSTM(50, input_shape=(2, 1)))
model.add(Dense(1))
model.compile(loss='mean_squared_error', optimizer='adam')
history = model.fit(x_train.reshape(-1, 2, 1), y_train, epochs=100, batch_size=32, validation_data=(x_test.reshape(-1, 2, 1), y_test))
plt.plot(history.history['loss'])
plt.plot(history.history['val_loss'])
plt.title('Model loss')
plt.ylabel('Loss')
plt.xlabel('Epoch')
plt.legend(['Train', 'Test'], loc='upper right')
plt.show()
train_predict = model.predict(x_train.reshape(-1, 2, 1))
test_predict = model.predict(x_test.reshape(-1, 2, 1))
train_predict = scaler.inverse_transform(train_predict)
train_predict = train_predict.reshape(-1) # 将结果变为一维数组
y_train = scaler.inverse_transform(y_train.reshape(-1, 1)).reshape(-1) # 将结果变为一维数组
test_predict = scaler.inverse_transform(test_predict)
y_test = scaler.inverse_transform(y_test.reshape(-1, 1)).reshape(-1) # 将结果变为一维数组
plt.plot(y_train, label='train')
plt.plot(train_predict, label='train predict')
plt.plot(y_test, label='test')
plt.plot(test_predict, label='test predict')
plt.legend()
plt.show()
```
import pandas as pd import numpy as np import matplotlib.pyplot as plt import tensorflow as tf from tensorflow.keras.models import Sequential from tensorflow.keras.layers import LSTM, Dense data = pd.read_csv('车辆:274序:4结果数据.csv') x = data[['车头间距', '原车道前车速度']].values y = data['本车速度'].values train_size = int(len(x) * 0.7) test_size = len(x) - train_size x_train, x_test = x[0:train_size,:], x[train_size:len(x),:] y_train, y_test = y[0:train_size], y[train_size:len(y)] from sklearn.preprocessing import MinMaxScaler scaler = MinMaxScaler(feature_range=(0, 1)) x_train = scaler.fit_transform(x_train) x_test = scaler.transform(x_test) model = Sequential() model.add(LSTM(50, input_shape=(2, 1))) model.add(Dense(1)) model.compile(loss='mean_squared_error', optimizer='adam') history = model.fit(x_train.reshape(-1, 2, 1), y_train, epochs=100, batch_size=32, validation_data=(x_test.reshape(-1, 2, 1), y_test)) plt.plot(history.history['loss']) plt.plot(history.history['val_loss']) plt.title('Model loss') plt.ylabel('Loss') plt.xlabel('Epoch') plt.legend(['Train', 'Test'], loc='upper right') plt.show() train_predict = model.predict(x_train.reshape(-1, 2, 1)) test_predict = model.predict(x_test.reshape(-1, 2, 1)) train_predict = scaler.inverse_transform(train_predict) train_predict = train_predict.reshape(-1, 1) y_train = scaler.inverse_transform([y_train]) test_predict = scaler.inverse_transform(test_predict) y_test = scaler.inverse_transform([y_test]) plt.plot(y_train[0], label='train') plt.plot(train_predict[:,0], label='train predict') plt.plot(y_test[0], label='test') plt.plot(test_predict[:,0], label='test predict') plt.legend() plt.show()报错Traceback (most recent call last): File "C:\Users\马斌\Desktop\NGSIM_data_processing\80s\lstmtest.py", line 42, in <module> train_predict = scaler.inverse_transform(train_predict) File "D:\python\python3.9.5\pythonProject\venv\lib\site-packages\sklearn\preprocessing\_data.py", line 541, in inverse_transform X -= self.min_ ValueError: non-broadcastable output operand with shape (611,1) doesn't match the broadcast shape (611,2)
这个错误提示是因为在使用 `scaler.inverse_transform()` 函数时,输入的 `train_predict` 和 `y_train` 的维度不匹配,导致无法进行运算。具体来说,`train_predict` 是一个形状为 `(611, 1)` 的二维数组,而 `y_train` 是一个形状为 `(1, 611)` 的二维数组,因此无法直接进行运算。
你可以尝试使用以下代码将 `y_train` 转换为一维数组,再进行运算:
```
train_predict = scaler.inverse_transform(train_predict)
train_predict = train_predict.reshape(-1) # 将结果变为一维数组
y_train = scaler.inverse_transform(y_train.reshape(-1, 1)).reshape(-1) # 将结果变为一维数组
```
同样的,你也需要将 `y_test` 转换为一维数组,再进行相应的操作。