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)