我尝试使用Keras实现LSTM来解决多类问题。我输入了尺寸为1007x5的csv。每个实例的特性数为5个,共有12个类。下面是代码seed = 7
numpy.random.seed(seed)
input_file = 'input.csv'
def load_data(test_split = 0.2):
print ('Loading data...')
dataframe = pandas.read_csv(input_file, header=None)
dataset = dataframe.values
X = dataset[:,0:5].astype(float)
print(X)
Y = dataset[:,5]
print("y=", Y)
return X,Y
def create_model(X):
print ('Creating model...')
model = Sequential()
model.add(LSTM(128, input_shape =(5,)))
model.add(Dense(12, activation='sigmoid'))
print ('Compiling...')
model.compile(loss='categorical_crossentropy',
optimizer='rmsprop',
metrics=['accuracy'])
return model
X