1 files changed, 2 insertions, 2 deletions
diff --git a/train.py b/train.py
index 1588751..a990c13 100644
--- a/train.py
+++ b/train.py
@@ -1,5 +1,5 @@
-import ensemble-train
-import cross-train
+import ensemble
+import cross
 
 clist=[0.0001, 0.005, 0.01, 0.05, 0.1, 0.5, 1, 10, 100, 1000]
 inr="a.rid"