【发布时间】:2018-08-29 20:48:14
【问题描述】:
class MNISTModel:
def __init__(self, restore, session=None):
self.num_channels = 1
self.image_size = 28
self.num_labels = 10
model = Sequential()
model.add(Conv2D(32, (3, 3),
input_shape=(28, 28, 1)))
model.add(Activation('relu'))
model.add(Conv2D(32, (3, 3)))
model.add(Activation('relu'))
model.add(MaxPooling2D(pool_size=(2, 2)))
model.add(Conv2D(64, (3, 3)))
model.add(Activation('relu'))
model.add(Conv2D(64, (3, 3)))
model.add(Activation('relu'))
model.add(MaxPooling2D(pool_size=(2, 2)))
model.add(Flatten())
model.add(Dense(200))
model.add(Activation('relu'))
model.add(Dense(200))
model.add(Activation('relu'))
model.add(Dense(10))
model.load_weights(restore)
self.model = model
print('selfMNIST')
def predict(self, data):
tmp=self.model(data) #Question is here
return tmp
这句话“tmp=self.model(data)”是干什么用的? 这里的“模型”是类顺序的变量,我从来没有见过这样的用法。 此代码取自 ## 版权所有 (C) 2016, Nicholas Carlini
【问题讨论】:
标签: python model keras sequential