因子分解机——代码

利用矩阵,梯度下降算法实现的因子分解机
 

import numpy as np

LEARNING_RATE = 0.95
REGULARIZATION_COEFFICIENT = 0.000001
STEP_THRESHOLD = 300
K = 6

def sigmoid(x):
    return 1/(1+np.exp(-x))

class factorizationMachine:
    #x: a sample in train_set
    #cross_v: it is a two-dimensional numpy, whose shape is (k,n).And n is sample's feature number.
    def __init__(self, w0, linear_w, cross_v):
        self.w0 = w0
        self.linear_w = linear_w
        self.cross_v = cross_v

    #def pi_map(self, i):
    
    def inference(self, x):
        return self.w0 + self.linear_w.dot(x) + np.sum((self.cross_v.dot(x))**2) - np.sum((self.cross_v**2).dot(x**2))
    
    def loss(self, y, x):
        # y_ is predicted value, y is real value
        y_ = self.inference(x)
        #L2 regularization
     

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值