Вы можете увидеть фрагменты кода ниже.
[основной файл]
Код: Выделить всё
print("test1")
def classification():
print("test2")
raw_text = str(entry.get("1.0", tk.END))
tahmin_sonucu = main(raw_text)
categories_label.config(text="Kategoriler: " + tahmin_sonucu)
Код: Выделить всё
from simpletransformers.classification import ClassificationModel
labels = ["Bilim Kurgu", "Ekonomi", "İslam", "Polisiye", "Romantik", "Sağlık", "Spor"]
model = ClassificationModel('bert', 'bert_model', num_labels=7, use_cuda=False,
args={'reprocess_input_data': True, 'overwrite_output_dir': True, 'num_train_epochs': 3,
"train_batch_size": 16, "fp16": False, "output_dir": "bert_model"})
print("test11")
def main(metin):
tahmin = None
print("tes12")
tahmin=model.predict([metin])
print(labels[tahmin[0][0]])
return labels[tahmin[0][0]]
Подробнее здесь: https://stackoverflow.com/questions/784 ... secutively