使用kears三元分类器MLP碰到的问题

最新推荐文章于 2024-09-01 21:33:12 发布

愿航

最新推荐文章于 2024-09-01 21:33:12 发布

阅读量412

点赞数

文章标签：深度学习

本文链接：https://blog.csdn.net/wish_to_top/article/details/107204661

版权

记录贴

一级标题

安装版本

首先是安装版本，看到博主说要安装配套的keras和tensorflow，不然后面一直报错，这边用的是1.30的tensorflow和2.0.8的keras

三元分类器

忘了从哪位大佬博主那儿找来的了，源代码如下
from keras.models import Sequential
from keras.layers import Dense, Dropout, Activation
from keras.optimizers import SGD

#生成随机数据
import numpy as np
x_train = np.random.random((1000, 20))
y_train = keras.utils.to_categorical(np.random.randint(10, size=(1000, 1)), num_classes=10)
x_test = np.random.random((100, 20))
y_test = keras.utils.to_categorical(np.random.randint(10, size=(100, 1)), num_classes=10)

model = Sequential()
#Dense(64) is a fully-connected layer with 64 hidden units.
#in the first layer, you must specify the expected input data shape:
#here, 20-dimensional vectors.
model.add(Dense(64, activation=‘relu’, input_dim=20))
model.add(Dropout(0.5))
model.add(Dense(64, activation=‘relu’))
model.add(Dropout(0.5))
model.add(Dense(10, activation=‘softmax’))

sgd = SGD(lr=0.01, decay=1e-6, momentum=0.9, nesterov=True)
model.compile(loss=‘categorical_crossentropy’,
optimizer=sgd,
metrics=[‘accuracy’])

model.fit(x_train, y_train,
epochs=20,
batch_size=128)
score = model.evaluate(x_test, y_test, batch_size=128)
y_train

修改版本

X_train=df_train_degShuffle
y_train=df_train_deg_labelShuffle
X_train=np.array(X_train)
y_train= keras.utils.to_categorical(y_train)
mlp_model=mlp_model_construct(X_train,y_train)
train_model=mlp_model.evaluate(X_train,y_train)