plt.plot(scaler.inverse_transform((model(x_train_tensor).detach().numpy()).reshape(-1, 1)), "b") plt.plot(scaler.inverse_transform(y_train_tensor.detach().numpy().reshape(-1, 1)), "r")中model(x_train_tensor有60000个数,怎么取前1000个
时间: 2024-02-28 09:57:59 浏览: 101
Numpy之reshape()使用详解
可以使用切片操作来获取前1000个数,如下所示:
```
plt.plot(scaler.inverse_transform((model(x_train_tensor)[:1000].detach().numpy()).reshape(-1, 1)), "b")
plt.plot(scaler.inverse_transform(y_train_tensor[:1000].detach().numpy().reshape(-1, 1)), "r")
```
这里使用了`model(x_train_tensor)[:1000]`来获取前1000个数,`y_train_tensor[:1000]`也是同样的操作。注意要在获取前1000个数后再进行转换和reshape操作。
阅读全文