From 7a536eabbd79cee5fab944ca7a507715dc0905f8 Mon Sep 17 00:00:00 2001
From: missQian <13240016260@163.com>
Date: Sun, 4 Oct 2020 21:00:58 +0800
Subject: [PATCH] Create train.py

---
 .../train.py                                  | 26 +++++++++++++++++++
 1 file changed, 26 insertions(+)
 create mode 100644 models/HMEAE: Hierarchical Modular Event Argument Extraction/train.py

diff --git a/models/HMEAE: Hierarchical Modular Event Argument Extraction/train.py b/models/HMEAE: Hierarchical Modular Event Argument Extraction/train.py
new file mode 100644
index 0000000..de49ccb
--- /dev/null
+++ b/models/HMEAE: Hierarchical Modular Event Argument Extraction/train.py	
@@ -0,0 +1,26 @@
+import tensorflow as tf
+import utils
+from models import DMCNN
+import os
+
+flags = tf.flags
+flags.DEFINE_string("gpu", "1", "The GPU to run on")
+flags.DEFINE_string("mode", "HMEAE", "DMCNN or HMEAE")
+flags.DEFINE_string("classify", "tuple", "single or tuple")
+
+def main(_):
+    config = flags.FLAGS
+    os.environ['CUDA_VISIBLE_DEVICES'] = config.gpu
+    extractor = utils.Extractor()
+    extractor.Extract()
+    loader = utils.Loader()
+    t_data = loader.load_trigger()
+    a_data = loader.load_argument()
+    trigger = DMCNN(t_data,a_data,loader.maxlen,loader.max_argument_len,loader.wordemb)
+    a_data_process = trigger.train_trigger()
+    argument = DMCNN(t_data,a_data_process,loader.maxlen,loader.max_argument_len,loader.wordemb,stage=config.mode,classify=config.classify)
+    argument.train_argument()
+
+
+if __name__=="__main__":
+    tf.app.run()