skopstask / README.md
tareknaser's picture
Upload 4 files
f487d06

Internship appplication task

Position: ML Open Source Engineer Internship - skops: Hugging Face and scikit-learn

Task requirments

  1. Create a python environment and install scikit-learn version 1.0 in that environment.
  2. Using that environment, create a LogisticRegression model and fit it on the Iris dataset.
  3. Save the trained model using pickle or joblib.
  4. Create a second environment, and install scikit-learn version 1.1 in it.
  5. Try loading the model you saved in step 3 in this second environment.

Steps Taken

  1. I used mamba to create the environment locally.
  2. Trained a simple logistic regression model in model.ipynb.
  3. Used pickle to save the model in finalized_model.sav.
  4. Created another environment with scikit-learn version 1.1.
  5. Imported the model in importingmodel.ipynb.

Observations

A warning is shown when trying to load the model.

/home/tarek/.local/lib/python3.8/site-packages/sklearn/base.py:329: UserWarning: Trying to unpickle estimator LogisticRegression from version 1.0.2 when using version 1.1.0. This might lead to breaking code or invalid results. Use at your own risk. For more info please refer to:
https://scikit-learn.org/stable/modules/model_persistence.html#security-maintainability-limitations
  warnings.warn(