def train(): print('Loading amr data') paragraphs = generate_paragraphs('amr.txt', k=5) print('%d total cleaned paragraphs' % len(paragraphs)) print('Training Subgraph Selection Scorer') train_instances, train_labels = gen_subgraph_data(paragraphs) subgraph_scorer = SubgraphSelectionScorer() subgraph_scorer.train(train_instances, train_labels, update_cache=True) print('Training Order Scorer') train_instances, train_labels, train_weights = gen_order_data(paragraphs) order_scorer = OrderScorer() order_scorer.train(train_instances, train_labels, train_weights) print('Training Pipeline Scorer') pipeline_scorer = PipelineScorer() subgraph_optimizer = SubgraphOptimizer(subgraph_scorer) order_optimizer = OrderOptimizer(order_scorer) pipeline_scorer.train(subgraph_optimizer, order_optimizer)