Class SVDPlusPlus

Object
org.apache.spark.graphx.lib.SVDPlusPlus

public class SVDPlusPlus extends Object
Implementation of SVD++ algorithm.
  • Constructor Details

    • SVDPlusPlus

      public SVDPlusPlus()
  • Method Details

    • run

      public static scala.Tuple2<Graph<scala.Tuple4<double[],double[],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 here.

      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