2021
02-21
02-21
TensorFlow2.0使用keras训练模型的实现
1.一般的模型构造、训练、测试流程#模型构造inputs=keras.Input(shape=(784,),name='mnist_input')h1=layers.Dense(64,activation='relu')(inputs)h1=layers.Dense(64,activation='relu')(h1)outputs=layers.Dense(10,activation='softmax')(h1)model=keras.Model(inputs,outputs)#keras.utils.plot_model(model,'net001.png',show_shapes=True)model.compile(optimizer=keras.optimizers.RMSprop(),loss=k...
继续阅读 >