def main(_):
#mode = "evalfreeze"
mode = "pretrain"
evaldata = "20171213142927.xls"
model = DomainNameModel(mode='freeze', learning_rate=0.0008)
worker = Work(model,
batch_size=1024,
pretrain_iter=20000,
train_iter=900000,
sample_iter=100,
model_save_path='model',
sample_path='sample',
plk_dir='./plk')
if mode == 'pretrain':
pretain()
else: #eval
dosample = preprosample(None, plk_dir='./plk/')
worker.evalfreeze(evaldata, dosample)
