kubeflow / fairing

Python SDK for building, training, and deploying ML models
Apache License 2.0
337 stars 144 forks source link

Which notebook image to use xgboost-high-level-apis example? #202

Open jlewi opened 5 years ago

jlewi commented 5 years ago

Is there an existing notebook image suitable for running the example:

https://github.com/kubeflow/fairing/blob/master/examples/prediction/xgboost-high-level-apis.ipynb

I tried using gcr.io/kubeflow-images-public/tensorflow-1.13.1-notebook-cpu:v0.5.0

But it looks like its missing some dependencies.

import argparse
import logging
import joblib
import sys
import pandas as pd
from sklearn.metrics import mean_absolute_error
from sklearn.model_selection import train_test_split
from sklearn.impute import SimpleImputer
from xgboost import XGBRegressor

I was able to work around this just by installing the modules.

!pip install joblib
!pip install pandas
!pip install scikit-learn

/cc @karthikv2k @r2d4

karthikv2k commented 5 years ago

you have to install the requirements in https://github.com/kubeflow/fairing/blob/master/examples/prediction/requirements.txt . We didn't create a separate image for this example. We will add "!pip install -r requirements.txt" as the first cell in the notebook to let the users know they have to install the deps.