稀疏矩阵 - 错误类型错误:“str”对象不能解释为整数

时间:2021-08-02 00:33:45

标签: python sparse-matrix recommendation-engine

晚安!

我正在尝试使用 Pandas 数据框创建一个稀疏数组,但出现以下错误。 我该如何解决这个问题?

treino_df.dtypes

enter image description here

我认为问题在于某些变量不是 int(而是 object)

if os.path.isfile('matriz_esparsa_treino.npz'):
    matriz_esparsa_treino = sparse.load_npz('matriz_esparsa_treino.npz')
    print("Matriz Carregada.")

else: 
    matriz_esparsa_treino = sparse.csr_matrix((treino_df.Nota.values, (treino_df.ID_Usuario.values, 
                                                                                 treino_df.ID_Produto.values)))
    print('Matriz Criada. O shape é: (user, movie): ', matriz_esparsa_treino.shape)
    sparse.save_npz("dados/matriz_esparsa_treino.npz", matriz_esparsa_treino)
    print('Matriz Salva em Disco.')

0 个答案:

没有答案