【发布时间】:2022-01-16 06:17:50
【问题描述】:
我正在处理我自己的数据集,该数据集存储在 csv 文件中。它有三列: val1 | val2 |标签。共有 6 个标签。行数和列数分别为 2000 和 3。我想创建一个 1D CNN 网络,它接受输入 val1 和 val2 并可以预测标签。到目前为止我已经尝试过
df = pd.read_csv("data.csv")
x = df.drop(["label"], axis=1) #x.shape = (2000, 2)
x = np.expand_dims(x,-1) #x.shape = (2000, 2, 1)
y = df.label #y.shape = (2000, 1)
y = to_categorical(y) #y.shape = (2000, 6)
X_train, X_test, y_train, y_test = train_test_split(x, y, test_size=0.2)
X_train, X_valid, y_train, y_valid = train_test_split(X_train, y_train, test_size=0.2)
model = Sequential()
model.add(Conv1D(filters=256, kernel_size=2, activation='relu', input_shape=(2,1)))
model.add(Dropout(0.2))
model.add(MaxPooling1D(pool_size=1))
model.add(Flatten())
model.add(Dense(128, activation='relu'))
model.add(Dropout(0.2))
model.add(Dense(6, activation='softmax'))
model.compile(loss='categorical_crossentropy', optimizer='adam', metrics=['accuracy'])
model.fit(X_train, y_train,
batch_size=64,
epochs=100,
verbose=1,
validation_data=(X_valid,y_valid),
shuffle=True,
)
上述模型仅提供最高 30% 的验证和训练准确度。
我尝试过的事情: 数据增强。 更改过滤器的数量。 增加层数。
如何提高模型的准确性?
【问题讨论】:
标签: tensorflow keras neural-network conv-neural-network