org.apache.spark.graphx.lib

Class SVDPlusPlus

Object

org.apache.spark.graphx.lib.SVDPlusPlus

public class SVDPlusPlus
extends Object

Implementation of SVD++ algorithm.

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	SVDPlusPlus.Conf Configuration parameters for SVDPlusPlus.

Constructor Summary

Constructors

Constructor and Description
SVDPlusPlus()

Method Summary

Methods

Modifier and Type	Method and Description
static scala.Tuple2<Graph<scala.Tuple4<org.jblas.DoubleMatrix,org.jblas.DoubleMatrix,Object,Object>,Object>,Object>	run(RDD<Edge<Object>> edges, SVDPlusPlus.Conf conf) Implement SVD++ based on "Factorization Meets the Neighborhood: a Multifaceted Collaborative Filtering Model", available at http://public.research.att.com/~volinsky/netflix/kdd08koren.pdf.

Methods inherited from class Object

equals, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

SVDPlusPlus

public SVDPlusPlus()

Method Detail

run

public static scala.Tuple2<Graph<scala.Tuple4<org.jblas.DoubleMatrix,org.jblas.DoubleMatrix,Object,Object>,Object>,Object> run(RDD<Edge<Object>> edges,
                                                                                                               SVDPlusPlus.Conf conf)

Implement SVD++ based on "Factorization Meets the Neighborhood: a Multifaceted Collaborative Filtering Model", available at http://public.research.att.com/~volinsky/netflix/kdd08koren.pdf.

The prediction rule is rui = u + bu + bi + qi*(pu + |N(u)|^(-0.5)*sum(y)), see the details on page 6.

Parameters:

edges - edges for constructing the graph

conf - SVDPlusPlus parameters

Returns:

a graph with vertex attributes containing the trained model