AttributeError: объект 'FlaubertForSequenceClassification' не имеет атрибута 'predict'

Я пытаюсь классифицировать новый набор данных на основе лучшей модели после загрузки, но получаю эту ошибку:

best_model_from_training_testing = './checkpoint-900'
best_model= FlaubertForSequenceClassification.from_pretrained(best_model_from_training_testing, num_labels=3)

raw_pred, _, _ = best_model.predict(test_tokenized_dataset)         
predictedLabelOnCompanyData = np.argmax(raw_pred, axis=1)

Traceback (most recent call last):
  File "/16uw/test/MODULE_FMC/scriptTraitements/classifying.py", line 408, in <module>
    pred = predict(emb, test_model)
  File "/16uw/test/MODULE_FMC/scriptTraitements/classifying.py", line 279, in predict
    raw_pred, _, _ = model.predict(emb_feature)
  File "/l16uw/.conda/envs/bert/lib/python3.9/site-packages/torch/nn/modules/module.py", line 1130, in __getattr__
    raise AttributeError("'{}' object has no attribute '{}'".format(
AttributeError: 'FlaubertForSequenceClassification' object has no attribute 'predict'

Вернуться на верх