Fix index error when using from_delayed: see https://github.com/tmoer…

…man/arboreto/issues/17
opoirion · Aug 22, 2019 · 2d5ac7c · 2d5ac7c
1 parent 3ff7b6f
commit 2d5ac7c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/arboreto/core.py b/arboreto/core.py
@@ -202,7 +202,7 @@ def pythonic():
 
         clean_links_df = links_df[links_df.importance > 0].sort_values(by='importance', ascending=False)
 
-        return clean_links_df[['TF', 'target', 'importance']]
+        return clean_links_df[['TF', 'importance', 'target']]
 
     if is_sklearn_regressor(regressor_type):
         return pythonic()