AttributeError: объект «implicit.evaluation._memoryviewslice» не имеет атрибута «dtype» ⇐ Python

[Anonymous]

Я обучил модель ALS, используя неявный пакет с поддержкой графического процессора. Однако при оценке модели с помощью функции ndcg_at_k я столкнулся со следующей ошибкой:

Код: Выделить всё

from implicit.evaluation import ndcg_at_k, mean_average_precision_at_k

ndcg = ndcg_at_k(model, train_user_items, test_user_items, K=14, show_progress=True, num_threads=1)

Код: Выделить всё

AttributeError: 'implicit.evaluation._memoryviewslice' object has no attribute 'dtype'

Сведения об среде:

[*]неявно: 0.7.2 (собрано из исходного кода)
[*]Python: 3.11.2
[*]CUDA: 12.3
[*]ОС: Debian GNU/Linux 12

Попытки решения:

• Я попробовал преобразовать модель в процессор, но это не решило проблему.

Вопрос:
Как устранить эту ошибку AttributeError при оценке модели с помощью неявной библиотеки?

Подробнее здесь: https://stackoverflow.com/questions/793 ... -attribute