def main():
if parameters.FULL_DATA_MODE:
lm = LanguageModel()
lm.load_data(train_data_path=parameters.SMALL_DATA_PATH)
lm.define_model()
lm.compile_model()
lm.fit_model()
lm.evaluate_model()
print(lm.generate_seq('女', 5))
else:
lm = LanguageModel()
lm.prepare_for_generator(train_data_path=parameters.TRAIN_DATA_PATH,
val_data_path=parameters.VAL_DATA_PATH,
test_data_path=parameters.TEST_DATA_PATH)
lm.define_model()
lm.compile_model()
lm.fit_model_with_generator()
lm.evaluate_model_with_generator()
