ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-06-23 01:55:38 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update joblib example to latest release

Browse Source
This commit is contained in:
allegroai 2019-12-24 18:21:10 +02:00
parent ba79471848
commit 865ad925dd
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
examples/joblib_example.py
View File

@ -1,7 +1,7 @@
try: try:
from sklearn.externals import joblib
except ImportError:
import joblib import joblib
except ImportError:
from sklearn.externals import joblib
from sklearn import datasets from sklearn import datasets
from sklearn.linear_model import LogisticRegression from sklearn.linear_model import LogisticRegression
@ -20,7 +20,7 @@ y = iris.target
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42) X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42)
model = LogisticRegression() # sklearn LogisticRegression class model = LogisticRegression(solver='liblinear', multi_class='auto') # sklearn LogisticRegression class
model.fit(X_train, y_train) model.fit(X_train, y_train)
joblib.dump(model, 'model.pkl', compress=True) joblib.dump(model, 'model.pkl', compress=True)