Allow plot function to handle XGBModel

2015-10-21 23:39:27 +09:00
parent eee3046624
commit 6f046327ac
3 changed files with 62 additions and 13 deletions
--- a/scripts/travis_script.sh
+++ b/scripts/travis_script.sh
@@ -64,7 +64,7 @@ if [ ${TASK} == "python-package" -o ${TASK} == "python-package3" ]; then
        conda create -n myenv python=2.7
    fi
    source activate myenv
-    conda install numpy scipy pandas matplotlib nose
+    conda install numpy scipy pandas matplotlib nose scikit-learn
    python -m pip install graphviz

    make all CXX=${CXX} || exit -1