Aula 08 - Salvando e importando o modelo



Para que possamos utilizar posteriormente o classificador treinado, devemos salvar o modelo através de um pickle file.

# salvando o modelo

with open('classifier.pickle', 'wb') as f:
    pickle.dump(classifier,f)


with open ('tfidfmodel.pickle', 'wb') as f: pickle.dump(vectorizer,f)


# unpickling the classifier and vectorizer

with open('classifier.pickle', 'rb') as f:
    clf = pickle.load(f)
    
with open('tfidfmodel.pickle', 'rb') as f:
    tfidf = pickle.load(f)
    
sample = ['you are a nice person man, have a good life']
sample = tfidf.transform(sample).toarray()
    
print (clf.predict(sample))











Aula 08 - Salvando e importando o modelo Aula 08 - Salvando e importando o modelo Reviewed by Vinicius dos Santos on 09:39:00 Rating: 5

Nenhum comentário

Escreve ai sua opinião!