FM算法的实现方式

AiBigData

于 2021-10-24 21:02:32 发布

阅读量2.6k

点赞数

CC 4.0 BY-SA版权

分类专栏：推荐算法实现文章标签： 1024程序员节算法 tensorflow

本文链接：https://blog.csdn.net/AiBigData/article/details/120940907

推荐算法实现专栏收录该内容

1 篇文章

订阅专栏

该博客介绍了TensorFlow中实现的因子分解机（Factorization Machine, FM）模型，这是一种用于特征交互的模型，特别是针对二阶交互。FM模型不包含线性项和偏置项。输入要求为3D张量，输出为2D张量。模型通过计算特征的平方和、平方差以及交叉项来捕捉特征之间的交互，最终返回交叉项的一半作为输出。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

class FM(Layer):
    """Factorization Machine models pairwise (order-2) feature interactions
     without linear term and bias.

      Input shape
        - 3D tensor with shape: ``(batch_size,field_size,embedding_size)``.

      Output shape
        - 2D tensor with shape: ``(batch_size, 1)``.

      References
        - [Factorization Machines](https://www.csie.ntu.edu.tw/~b97053/paper/Rendle2010FM.pdf)
    """

    def __init__(self, **kwargs):

        super(FM, self).__init__(**kwargs)

    def build(self, input_shape):
        if len(input_shape) != 3:
            raise ValueError("Unexpected inputs dimensions % d,\
                             expect to be 3 dimensions" % (len(input_shape)))

        super(FM, self).build(input_shape)  # Be sure to call this somewhere!

    def call(self, inputs, **kwargs):

        if K.ndim(inputs) != 3:
            raise ValueError(
                "Unexpected inputs dimensions %d, expect to be 3 dimensions"
                % (K.ndim(inputs)))

        concated_embeds_value = inputs

        square_of_sum = tf.square(reduce_sum(
            concated_embeds_value, axis=1, keep_dims=True))
        sum_of_square = reduce_sum(
            concated_embeds_value * concated_embeds_value, axis=1, keep_dims=True)
        cross_term = square_of_sum - sum_of_square
        cross_term = 0.5 * reduce_sum(cross_term, axis=2, keep_dims=False)

        return cross_term

    def compute_output_shape(self, input_shape):
        return (None, 1)