[BUG] 由于 numpy 对 ndim > 1 的向量转换进行了弃用,导致 tfidf_utils.py 存在问题
作者: Baccano01创建于 2026年6月19日更新于 2026年6月19日
标签bug
描述
tfidf_utils.py 中的一个小错误,当在本地机器上运行 tfidf_covid.ipynb 笔记本时,特别是获取前 k 个推荐项:
cols_to_keep = ['title','authors','journal','publish_time','url']
recommender.get_top_k_recommendations(metadata_public,article_of_interest,cols_to_keep)
由于 tfidf 模型中的代码: https://GitHub.com/recommenders-team/recommenders/blob/6232b154548c955315650d58dca6bf1411c56020/recommenders/models/tfidf/tfidf_utils.py#L330 由于在 numpy 1.25 中转换 ndim > 0 数组为标量的功能已被弃用;
np.where(metadata[self.id_col] == rec_id)[0] 返回 ndim 为 1 的数组。
内容来源: recommenders-team/recommenders