算法模型保存与使用

在算法建立模块中进行模型保存

模型保存如下：

import joblib

counter = joblib.load('counter_model') # 加载已建立好的词袋模型
nb = joblib.load('nb_model') # 加载已建立好的算法模型


def new_text_pre():
    '''
    :param filepath:
    :return: 返回分类结果列表
    '''
    new_text = read_text('./text_classification-master/text classification/new_text', encoding='utf-8')
    new_text['text'] = cut_word(new_text['text'])
    new_text_x = counter.transform(new_text['text']).toarray()
    map_dict = {0: '体育', 1: '女性', 2: '文学', 3: '校园'}
    print(f'预测结果为：{[map_dict[i] for i in nb.predict(new_text_x)]}')


if __name__ == '__main__':
    new_text_pre()