def __init__(self, shape, optimizer='seq', lr=0.01, clip_min=0, clip_max=1081, is_round=True):
'''
param shape: phi_x shape
'''
self.optimizer = optimizer
self.lr = lr
self.shape = shape
self.learning_rate_decay = 0.99
# w is Mx1 column vector
self.w = np.asmatrix(self._init_weight(shape + (1,)))
# Setup model
self._setup_model()
# Setup optimizer (vary from models)
self._setup_optimizer()
评论列表
文章目录