def __init__(self, n_input, n_hidden, activation_func='softplus',
optimizer_name='AdamOptimizer',
learning_rate=0.001,
logdir='/tmp',
log_every_n=100,
session_kwargs={},
seed=42,
tied_weights=False,
linear_decoder=True,
):
'''
params:
activation_func (string): a name of activation_func in tf.nn
optimizer_name (string): a name of the optimizer object name tf.train
'''
self.n_input = n_input
self.n_hidden = n_hidden
self.activation_func = activation_func
self.optimizer_name = optimizer_name
self.learning_rate = learning_rate
self.logdir = logdir
self.log_every_n = log_every_n
self.session_kwargs = session_kwargs
self.seed = seed
self.tied_weights = tied_weights
self.linear_decoder = linear_decoder
self._init_all()
评论列表
文章目录