Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Enhance complete_analysis with model saving and add model loading functionality #61

Open
wants to merge 3 commits into
base: main
Choose a base branch
from

Conversation

WladRamos
Copy link

This Pull Request introduces two key improvements:

  1. Model Saving in complete_analysis:

    • The complete_analysis function has been enhanced with the ability to save the trained model to a .pkl file. This allows users to preserve models after training for later use without retraining.
    • The user can specify the filename for saving the model or leave it blank to use a default name with a timestamp.
    • The model is saved conditionally, based on the save_model parameter.
  2. Function to Load a Saved Model:

    • A new function load_model was added to load a model from a .pkl file. This allows users to reload the saved model and continue making predictions or further analysis.
  3. Unit Tests:

    • Added tests to ensure that the model is saved correctly and can be loaded without errors.
    • Verified that the loaded model retains its functionality and can make predictions with the test data.
    • Ensured proper cleanup of the saved model file after tests.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant