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) This method is deprecated in favor of runSVDPlusPlus(), which replaces DoubleMatrix with Array[Double] in its return value.
static scala.Tuple2<Graph<scala.Tuple4<double[],double[],Object,Object>,Object>,Object>	runSVDPlusPlus(RDD<Edge<Object>> edges, SVDPlusPlus.Conf conf) :: Experimental ::

Methods inherited from class Object

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

Constructor Detail

SVDPlusPlus

public SVDPlusPlus()

Method Detail

runSVDPlusPlus

public static scala.Tuple2<Graph<scala.Tuple4<double[],double[],Object,Object>,Object>,Object> runSVDPlusPlus(RDD<Edge<Object>> edges,
                                                                                              SVDPlusPlus.Conf conf)

:: Experimental ::

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.

This method temporarily replaces run(), and replaces DoubleMatrix in run()'s return value with Array[Double]. In 1.4.0, this method will be deprecated, but will be copied to replace run(), which will then be undeprecated.

Parameters:

edges - edges for constructing the graph

conf - SVDPlusPlus parameters

Returns:

a graph with vertex attributes containing the trained model

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)

This method is deprecated in favor of runSVDPlusPlus(), which replaces DoubleMatrix with Array[Double] in its return value. This method is deprecated. It will effectively be removed in 1.4.0 when runSVDPlusPlus() is copied to replace run(), and hence the return type of this method changes.