#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
item_based CF
"""
import numpy as np
import pandas as pd
class CFItem(object):
def __init__(self, k):
self.top_k = k
@staticmethod
def simu_cal_item(data_beha, i, j):
data_beha = np.array(data_beha)
data_beha = data_beha[:, [i, j]]
index = np.where(data_beha == 0)[0]
data_beha_a = np.delete(data_beha, obj=index, axis=0)
if data_beha_a.size == 0:
return 0
else:
data_beha_i = data_beha_a[:, 0]
data_beha_j = data_beha_a[:, 1]
return np.corrcoef(x=data_beha_i, y=data_beha_j)[0, 1]
def get_simu_mat(self, data_beha):
data_beha = np.array(data_beha)
mat = np.zeros([data_beha.shape[1], data_beha.shape[1]])
for i in range(mat.shape[0]):
for j in range(mat.shape[1]):
mat[i][j] = self.simu_cal_item(data_beha, i, j)
mat[j][i] = mat[i][j]
item-based CF(基于物品的协同过滤算法)
最新推荐文章于 2025-05-19 16:54:25 发布