sklearn_model_save_load_with_skops_secure_persistence.py

1import numpy as np
2from sklearn.linear_model import LogisticRegression
3from skops import io
4
5# 1. Prepare data and train a model
6X = np.array([[1, 1], [1, 2], [2, 2], [2, 3]])
7y = np.array([0, 0, 1, 1])
8clf = LogisticRegression().fit(X, y)
9
10# 2. Save the model using skops (safer alternative to pickle)
11file_name = "example.skops"
12io.dump(clf, file_name)
13
14# 3. Load the model
15# When loading, you must explicitly trust the types to ensure security
16unknown_types = io.get_untrusted_types(file=file_name)
17loaded_clf = io.load(file_name, trusted=unknown_types)
18
19# 4. Use the loaded model
20res = loaded_clf.predict(X)
21print(res)