From ed8942a0969edb0281ec39541efaa366376829dc Mon Sep 17 00:00:00 2001
From: Matthew Honnibal <honnibal@gmail.com>
Date: Wed, 8 Apr 2015 22:47:59 +0200
Subject: [PATCH] * Add train function to fabfile

---
 fabfile.py | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/fabfile.py b/fabfile.py
index 9a2236e77..dc6177107 100644
--- a/fabfile.py
+++ b/fabfile.py
@@ -56,6 +56,19 @@ def test():
             local('py.test -x')
 
 
+def train(train_loc=None, dev_loc=None, model_dir=None):
+    if train_loc is None:
+        train_loc = 'corpora/en/ym.wsj02-21.conll'
+    if dev_loc is None:
+        dev_loc = 'corpora/en/ym.wsj24.conll'
+    if model_dir is None:
+        model_dir = 'models/en/'
+    with virtualenv(VENV_DIR):
+        with lcd(path.dirname(__file__)):
+            local('python bin/init_model.py lang_data/en/ corpora/en/ ' + model_dir)
+            local('python bin/parser/train.py %s %s %s' % (train_loc, dev_loc, model_dir))
+
+
 def travis():
     local('open https://travis-ci.org/honnibal/thinc')