簡體   English   中英

向量的一維卷積

[英]1d convolution for vector

我正在學習了解如何使用具有一維卷積的卷積神經網絡:

這是一個家庭作業示例:

import numpy as np
import keras
from keras.models import Sequential
from keras.layers import Dense, Dropout, Flatten
from keras.layers import Conv2D, MaxPooling2D
from keras import backend as K
from keras.utils import np_utils
from keras.layers.convolutional import Conv1D
from keras.layers.convolutional import MaxPooling1D


epochs=20
batch_size=50
num_classes=20


x_train = np.random.rand(60000,400)
x_val = np.random.rand(10000,400)

y_tain = np.eye(20)[np.random.choice(5, 60000)]
y_val = np.eye(20)[np.random.choice(5, 10000)]

model = Sequential()
model.add(Conv1D(filters=64, kernel_size=3, activation='relu', input_shape=(400,)))
model.add(Conv1D(filters=64, kernel_size=3, activation='relu'))
model.add(Dropout(0.5))
model.add(MaxPooling1D(pool_size=2))
model.add(Flatten())
model.add(Dense(100, activation='relu'))
model.add(Dense(20, activation='softmax'))
model.compile(loss='categorical_crossentropy', optimizer='adam', metrics=['accuracy'])


model.fit(x_train, y_tain,
          batch_size=batch_size,
          epochs=epochs,
          verbose=1,
          validation_data=(x_val, y_val))
score = model.evaluate(x_val, y_val, verbose=0)
print('Test loss:', score[0])
print('Test accuracy:', score[1])

當我嘗試運行它時,出現了一些錯誤:

ValueError: Input 0 is incompatible with layer conv1d_1: expected ndim=3, found ndim=2

你如何讓它編譯?

仔細查看錯誤消息:您的輸入是二維的,而您的卷積層需要三個維度...

暫無
暫無

聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.

 
粵ICP備18138465號  © 2020-2024 STACKOOM.COM