from tensorflow.keras import Sequential
from tensorflow.keras.layers import Flatten,Dense,Dropout
import numpy as np
# 模型定义
model = Sequential([
Flatten(input_shape=(28, 28)),
Dense(units=200,activation='tanh'),
Dropout(0.4),
Dense(units=100,activation='tanh'),
Dropout(0.4),
Dense(units=10,activation='softmax')
])
/*
开发不易，整理也不易，如需要详细的说明文档和程序，以及完整的数据集，训练好的模型，或者进一步开发，
可加作者新联系方式咨询，WX：Q3101759565，QQ：3101759565
*/
from tensorflow.keras import Sequential
from tensorflow.keras.layers import Flatten,Dense,Dropout
import numpy as np
# 模型结构定义 (784,200,100,10)
model = Sequential([
Flatten(input_shape=(28, 28)),
Dense(units=200,activation='tanh'),
Dropout(0.4),
Dense(units=100,activation='tanh'),
Dropout(0.4),
Dense(units=10,activation='softmax')
])
# 载入模型参数 model_weights.h5
model.load_weights('./my_model/model_weights.h5')
print(model.summary())
weights = model.get_weights()
weights = np.array(weights)
for i,w in enumerate(weights):
if i % 2 == 0:
print('{}:w_shape:{}'.format(int(i / 2 + 1), w))
else:
print('{}:b_shape:{}'.format(int(i / 2 + 0.5), w.shape))
