Skip to content

Commit

Permalink
Update examples
Browse files Browse the repository at this point in the history
  • Loading branch information
polyaxon-team committed Oct 15, 2020
1 parent 2852dc6 commit 4f7bf55
Show file tree
Hide file tree
Showing 3 changed files with 6 additions and 2 deletions.
2 changes: 2 additions & 0 deletions in_cluster/build_image/build.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -23,5 +23,7 @@ runPatch:
- pip3 install tensorboard
- pip3 install matplotlib
- pip3 install fastai
- pip3 install vega-datasets
- pip3 install joblib
langEnv: 'en_US.UTF-8'
hubRef: kaniko
3 changes: 2 additions & 1 deletion in_cluster/sklearn/iris/model.py
Original file line number Diff line number Diff line change
@@ -1,8 +1,9 @@
import joblib

from sklearn.model_selection import train_test_split
from sklearn.neighbors import KNeighborsClassifier
from sklearn import metrics
from sklearn.datasets import load_iris
from sklearn.externals import joblib


def train_and_eval(
Expand Down
3 changes: 2 additions & 1 deletion in_cluster/sklearn/newsgroup/model.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,8 +2,9 @@

from time import time

import joblib

from sklearn.datasets import fetch_20newsgroups
from sklearn.externals import joblib
from sklearn.feature_extraction.text import TfidfVectorizer
from sklearn.linear_model import LogisticRegression
from sklearn.pipeline import Pipeline
Expand Down

0 comments on commit 4f7bf55

Please sign in to comment.