【发布时间】:2022-02-20 20:11:31
【问题描述】:
x_train1, x_test, y_train1, y_test = train_test_split(images, labels,test_size=0.2,random_state=42)
x_train2, x_val,y_train2,y_val = train_test_split(x_train1, y_train1,test_size=0.05,random_state=42)
图层
model = Sequential()
model.add(Conv2D(32, (3, 3), activation = \'relu\', input_shape=(128,128,1), kernel_regularizer=keras.regularizers.l2(0.005), padding =\'same\', name=\'Conv_1\'))
model.add(MaxPooling2D((2,2),name=\'MaxPool_1\'))
model.add(Conv2D(64, (3, 3), activation = \'relu\',padding =\'same\', kernel_regularizer=keras.regularizers.l2(0.005), name=\'Conv_2\'))
model.add(MaxPooling2D((2,2),name=\'MaxPool_2\'))
model.add(Flatten(name=\'Flatten\'))
model.add(Dropout(0.5,name=\'Dropout\'))
model.add(Dense(64, kernel_initializer=\'normal\', activation=\'relu\', name=\'Dense_1\'))
model.add(Dense(1, kernel_initializer=\'normal\', activation=\'sigmoid\', name=\'Dense_2\'))
model.summary()
模型编译
model.compile(loss=\'binary_crossentropy\', optimizer=\'adam\', metrics=[\'accuracy\'])
history = model.fit(x_train2, y_train2,validation_data=(x_test, y_test),batch_size=32, epochs=100 )
** 结果 ** 训练:准确率 = 0.939577;损失 = 0.134506 测试:准确度 = 0.767908;损失 = 0.8002433
标签: machine-learning deep-learning neural-network conv-neural-network overfitting-underfitting