数组[:,0:1]=数组[:][0:1]?

2024-03-29 14:54:32 发布

您现在位置:Python中文网/ 问答频道 /正文

我在跟踪一个LSTM tutorial。你知道吗

为了将训练数据分为输入(x)和输出(y),我必须执行以下命令:

X, y = train[:, 0: -1], train[:, -1]

很遗憾,它无法工作,并且在打印train[:, 0: -1]时生成以下错误:

> TypeError: '(slice(None, None, None), slice(0, -1, None))' is an invalid key

我已尝试将此命令替换为:

X, y = train[:][0: -1], train[:][-1]

但我很肯定它不会给出相同的输出,因为有几个输入和一个输出是不合逻辑的(在我的例子中)。你知道吗

下面是一个最小的可复制代码和一个数据示例:

from pandas import DataFrame
from pandas import datetime
from keras.models import Sequential
from keras.layers import Dense
from keras.layers import LSTM
import numpy

O = [0.701733664614, 0.699495411782, 0.572129320819, 0.613315597684, 0.58079660603, 0.596638918579, 0.48453382119]
Ab = [datetime(2018, 12, 11, 14, 0), datetime(2018, 12, 21, 10, 0), datetime(2018, 12, 21, 14, 0), datetime(2019, 1, 1, 10, 0), datetime(2019, 1, 1, 14, 0), datetime(2019, 1, 11, 10, 0), datetime(2019, 1, 11, 14, 0)]

data = DataFrame(numpy.column_stack([O, Ab]),
                 columns=['ndvi', 'datetime'])

def fit_lstm(train, batch_size, nb_epoch, neurons):
    X, y = train[:, 0: -1], train[:, -1]
    X = X.values.reshape(X.shape[0], 1, X.shape[1])
    model = Sequential()
    model.add(LSTM(neurons, batch_input_shape=(batch_size, X.shape[1], X.shape[2]), stateful=True))
    model.add(Dense(1))
    model.compile(loss='mean_squared_error', optimizer='adam')
    for i in range(nb_epoch):
        model.fit(X, y, epochs=1, batch_size=batch_size, verbose=0, shuffle=False)
        model.reset_states()
    return model

train, test = data.values[0:-2], data.values[-2:]

print (train[:, 0:-1])

我只想解决这个问题以适应LSTM模型:

lstm_model = fit_lstm(train, 1, 3000, 4)

也许,在这种情况下,我必须使用shift(),将最后一个时间步作为输入,将当前一个作为输出? 像这样:

shift_steps = 1
train_targets = train.shift(-shift_steps)
X, y = train, train_targets

Tags: fromimportnonedatasizedatetimemodelshift
2条回答

我想我发现了我的错误: 语法与我的数据集不兼容。你知道吗

正如我在上次编辑问题时提到的,我使用了shift。它是这样工作的:


def fit_lstm(train, batch_size, nb_epoch, neurons):
    X, y = train, train_targets
    X = X.values.reshape(X.shape[0], 1, X.shape[1])
    model = Sequential()
    model.add(LSTM(neurons, batch_input_shape=(batch_size, X.shape[1], X.shape[2]), stateful=True))
    model.add(Dense(1))
    model.compile(loss='mean_squared_error', optimizer='adam')
    for i in range(nb_epoch):
        model.fit(X, y, epochs=1, batch_size=batch_size, verbose=0, shuffle=False)
        model.reset_states()
    return model

train, test = data.values[0:-2], data.values[-2:]
shift_steps = 1
train_targets = data.shift(-shift_steps).values[0:-2] 

| |

| |

/

print (train)
[0.70173366 0.69949541 0.57212932 0.6133156  0.58079661]
print (train_targets)
[0.69949541 0.57212932 0.6133156  0.58079661 0.59663892]

非常感谢@00和@Sebas的帮助:)

您使用的是DataFrame,切片是不同的。你知道吗

假设列“nvdi”包含这些特性,列datetime包含每个训练数据的预期结果,则需要将它们称为:

X = train['nvdi']
y = train['datetime']

我只看到数据中的两列。你知道吗

相关问题 更多 >