def train_nn_wavg_pr_model(embedding_method, dataset_id, layer, device_id):
data = load_train_data(dataset_id, 'classification')
train, val = stratified_sampling(data)
print(len(train), len(val))
trainer = ModelTrainer(embedding_method, dataset_id, layer, device_id)
model = trainer.train_nn_wavg_pr_model(train, val)
save_model(embedding_method, dataset_id, layer, 'nn_wavg_pr_model', model)
def train_nn_rouge_reg_model(embedding_method, dataset_id, layer, device_id):
data = load_train_data(dataset_id, 'regression_rouge')
train, val = stratified_sampling(data)
print(len(train), len(val))
trainer = ModelTrainer(embedding_method, dataset_id, layer, device_id)
model = trainer.train_nn_rouge_reg_model(train, val)
save_model(embedding_method, dataset_id, layer, 'nn_rouge_reg_model',
model)