Блог

С помощью scikit-learn я хотел бы предсказать 3 лучшие метки, связанные с текстом.

В FastText я решил проблему прямо так:

 model=ft.load_model('../model/icon_1307.bin')

...

#line by line, you make the predictions and store them in a list
predictions=[]
for line in df['subject']:
    pred_label=model.predict(line, k=3, threshold=0.001)  
    predictions.append(pred_label)

 

Согласно документации в Scikit, модуль sklearn.metrics.top_k_accuracy_score должен выполнять ту же работу. Но тогда, как я могу использовать свою модель и сделать прогноз в соответствии с ней?

Спасибо