lstm 输入数据维度_LSTM Keras值输入维度

我尝试使用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

  • 0
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值