skops_sklearn_model_save_load_with_trusted_types.py

1import skops.io as sio
2from sklearn.datasets import load_iris
3from sklearn.linear_model import LogisticRegression
4
5# Load data and train a model
6X, y = load_iris(return_X_y=True)
7clf = LogisticRegression(random_state=0, max_iter=1000).fit(X, y)
8
9# Save the model using skops
10sio.dump(clf, "model.skops")
11
12# To load the model, we first need to identify any untrusted types.
13# In a controlled environment where you trust the source, 
14# you can pass the list of types to the 'trusted' argument.
15unknown_types = sio.get_untrusted_types(file="model.skops")
16loaded_clf = sio.load("model.skops", trusted=unknown_types)
17
18# Verify the loaded model works
19print(loaded_clf.predict(X[:2]))