Пример #1
0
def test_with_labeled_state(data_filename,replacer, classifier, bow,tfidf):
    data = pd.read_csv(data_filename)

    bow_transformer = joblib.load(bow)
    tfidf_transformer = joblib.load(tfidf)
    classifier = joblib.load(classifier)
    data_labels = data['state'].tolist()
    data_texts = data['content']
    #preprocess data
    worker = Worker(bow = bow_transformer,tfidf = tfidf_transformer)
    word_list = preprocess(data_texts, replacer)
    text_vectors_list = worker.content_to_vectors(word_list)
    #test model
    pred = classifier.predict(text_vectors_list)
    worker.calculate_result(data_labels,pred)