File size: 623 Bytes
213a6de
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
import pickle
import numpy as np

def load_model():
    """
    Loads the trained model from file.
    """
    # load the vectorizer
    with open('./clf_NaiveBaised.pkl', 'rb') as fd:
        model = pickle.load(fd)

    return model

def model_predict(features):
    """
    Predicts using the loaded model.
    """
    model = load_model() # Hint: Load the model before predicting
    prediction =  model.predict([features])# Hint: Use the correct method to make predictions
    # If the email is spam, prediction should be 1, otherwise -1
    prediction =  'ham' if prediction[0] == 0 else 'spam'
    return prediction