Saving and loading models in ML Python - Cheat Sheet & Quick Revision

Recall & Review

beginner

What is the main purpose of saving a machine learning model?

Saving a model lets you keep the trained model so you can use it later without retraining. It's like saving your work in a game to continue playing later.

beginner

Name two common file formats used to save machine learning models.

Two common formats are:
1. Pickle (.pkl) for Python objects
2. HDF5 (.h5) often used with Keras models

intermediate

What is the difference between saving a model's architecture and saving its weights?

Saving architecture means saving the model's design (layers and connections). Saving weights means saving the learned numbers inside the model. Both are needed to fully restore the model.

beginner

Why is it important to load a saved model correctly before making predictions?

Loading the model correctly ensures it has the same structure and learned information. Otherwise, predictions might be wrong or cause errors.

intermediate

In TensorFlow/Keras, which function saves the entire model including architecture, weights, and optimizer state?

The function is model.save('filename'). It saves everything needed to reload and continue training or predicting.

What does saving a machine learning model allow you to do?

A Automatically improve model accuracy

B Make the model train faster

C Reuse the trained model without retraining

D Delete the training data

Saving a model lets you reuse it later without retraining, saving time and resources.

Which file format is commonly used to save Keras models?

A .h5

B .csv

C .txt

D .json

Keras models are often saved in HDF5 format with the .h5 extension.

What must you do before using a saved model to make predictions?

A Retrain the model

B Load the saved model correctly

C Change the model architecture

D Delete the saved file

You need to load the saved model correctly to use it for predictions.

Which Python library function is used to save a model as a pickle file?

A pickle.dump()

B json.save()

C model.save()

D numpy.save()

pickle.dump() saves Python objects like models to a file.

What does loading a model restore?

A Only the model's architecture

B Only the model's weights

C Only the training data

D Both architecture and weights

Loading a model restores both its architecture and learned weights.

Explain why saving and loading models is important in machine learning projects.

Think about how you would use a model after training it once.

Describe the steps to save a trained model and then load it back for making predictions.

Consider the functions or methods you might use in Python or TensorFlow.

Practice

1. What is the main purpose of saving a trained machine learning model?

easy

A. To increase the model's training speed

B. To reuse the model later without retraining

C. To improve the model's accuracy automatically

D. To delete the training data after use

Solution

1. Step 1: Understand model saving

   Saving a model stores its learned parameters so it can be reused later.

2. Step 2: Identify the main benefit

   This avoids retraining, saving time and resources.

3. Final Answer:

   To reuse the model later without retraining -> Option B

4. Quick Check:

   Saving = reuse without retraining [OK]

Hint: Saving means keep model for later use without retraining [OK]

Common Mistakes:

• Thinking saving improves accuracy automatically
• Confusing saving with speeding up training
• Believing saving deletes data

2. Which of the following is the correct way to save a Keras model named model to a file called my_model.h5?

easy

A. model.save('my_model.h5')

B. save_model('my_model.h5', model)

C. model.save_model('my_model.h5')

D. save('model', 'my_model.h5')

Solution

1. Step 1: Recall Keras save syntax

   Keras models are saved using model.save(filename).

2. Step 2: Match correct syntax

   model.save('my_model.h5') matches this syntax exactly.

3. Final Answer:

   model.save('my_model.h5') -> Option A

4. Quick Check:

   model.save(filename) saves model [OK]

Hint: Use model.save('filename') to save Keras models [OK]

Common Mistakes:

• Using incorrect function names like save_model or save
• Swapping argument order
• Calling save_model as a method on model

3. What will be the output of the following code snippet?

from tensorflow.keras.models import load_model
model = load_model('saved_model.h5')
predictions = model.predict([[5.1, 3.5, 1.4, 0.2]])
print(predictions)

medium

A. The model architecture printed as text

B. An error because load_model requires a directory, not a file

C. A list of class labels as strings

D. A numpy array of prediction probabilities

Solution

1. Step 1: Understand load_model usage

   load_model loads a saved Keras model from a file.

2. Step 2: Predict method output

   model.predict returns a numpy array of prediction probabilities for input data.

3. Final Answer:

   A numpy array of prediction probabilities -> Option D

4. Quick Check:

   load_model + predict = numpy array output [OK]

Hint: load_model loads model; predict returns probabilities array [OK]

Common Mistakes:

• Thinking load_model only loads from directories
• Expecting printed model architecture from predict
• Assuming predict returns class labels directly

4. You try to load a saved model using load_model('model.h5') but get an error: OSError: Unable to open file (file signature not found). What is the most likely cause?

medium

A. The model file is too large to load

B. You forgot to import load_model from keras

C. The file 'model.h5' is corrupted or not a valid model file

D. The model was saved with a different Python version

Solution

1. Step 1: Analyze the error message

   OSError about file signature means the file is not recognized as a valid HDF5 model file.

2. Step 2: Identify common causes

   This usually happens if the file is corrupted or not a saved model file.

3. Final Answer:

   The file 'model.h5' is corrupted or not a valid model file -> Option C

4. Quick Check:

   File signature error = corrupted or invalid file [OK]

Hint: File signature errors mean corrupted or wrong file type [OK]

Common Mistakes:

• Assuming Python version mismatch causes this error
• Forgetting to import but error message differs
• Thinking file size causes signature errors

5. You have a trained model saved as model_v1.h5. You want to load it, continue training on new data, then save the updated model as model_v2.h5. Which code snippet correctly does this?

hard

A. from tensorflow.keras.models import load_model
model = load_model('model_v1.h5')
model.fit(new_data, new_labels, epochs=5)
model.save('model_v2.h5')

B. model = load('model_v1.h5')
model.train(new_data, new_labels)
save_model(model, 'model_v2.h5')

C. load_model('model_v1.h5')
train_model(new_data, new_labels)
save('model_v2.h5')

D. from keras import load
model = load('model_v1.h5')
model.fit(new_data, new_labels)
model.save('model_v2.h5')

Solution

1. Step 1: Load the saved model correctly

   Use load_model('model_v1.h5') from tensorflow.keras.models to load the model.

2. Step 2: Continue training and save updated model

   Call model.fit() with new data, then save with model.save('model_v2.h5').

3. Final Answer:

   from tensorflow.keras.models import load_model model = load_model('model_v1.h5') model.fit(new_data, new_labels, epochs=5) model.save('model_v2.h5') -> Option A

4. Quick Check:

   Load, fit, save = from tensorflow.keras.models import load_model model = load_model('model_v1.h5') model.fit(new_data, new_labels, epochs=5) model.save('model_v2.h5') [OK]

Hint: Load with load_model, fit new data, then save again [OK]

Common Mistakes:

• Using undefined functions like load or save_model
• Forgetting to import load_model
• Not calling fit before saving updated model