blob: 4bde5307e02dc3d7a4cb115a789642494dc9679e [file] [log] [blame]
#!/bin/sh
###############################################
# Marvin AutoML installation Script #
###############################################
cmd=(dialog --title "Marvin AutoML" --separate-output --checklist "Select tools:" 22 76 16)
options=(1 "auto-sklearn" off # any option can be set to default to "on"
2 "h2o AutoML" off
3 "TPOT" off)
choices=$("${cmd[@]}" "${options[@]}" 2>&1 >/dev/tty)
clear
for choice in $choices
do
case $choice in
1)
echo "Instaling auto-sklearn..."
wget https://raw.githubusercontent.com/automl/auto-sklearn/master/requirements.txt \
| xargs -n 1 -L 1 pip install
pip install auto-sklearn
;;
2)
echo "Installing h2o..."
pip install requests
pip install tabulate
pip install scikit-learn
pip install http://h2o-release.s3.amazonaws.com/h2o/rel-yau/3/Python/h2o-3.26.0.3-py2.py3-none-any.whl
wget http://h2o-release.s3.amazonaws.com/h2o/rel-yau/3/h2o-3.26.0.3.zip
unzip h2o-3.26.0.3.zip
rm h2o-3.26.0.3.zip
;;
3)
echo "Installing TPOT..."
pip install tpot
;;
esac
done