io.kgraph.library.cf

Class Svdpp.UserComputation

java.lang.Object

io.kgraph.library.cf.Svdpp.UserComputation

All Implemented Interfaces:

ComputeFunction<CfLongId,Svdpp.SvdppValue,Float,FloatMatrixMessage>

Enclosing class:

Svdpp

public class Svdpp.UserComputation
extends Object
implements ComputeFunction<CfLongId,Svdpp.SvdppValue,Float,FloatMatrixMessage>

Nested Class Summary

Nested classes/interfaces inherited from interface io.kgraph.pregel.ComputeFunction

ComputeFunction.Aggregators, ComputeFunction.Callback<K,VV,EV,Message>, ComputeFunction.InitCallback, ComputeFunction.MasterCallback, ComputeFunction.ReadAggregators, ComputeFunction.ReadWriteAggregators

Constructor Summary

Constructors

Constructor and Description
UserComputation()

Method Summary

Modifier and Type	Method and Description
void	compute(int superstep, VertexWithValue<CfLongId,Svdpp.SvdppValue> vertex, Iterable<FloatMatrixMessage> messages, Iterable<EdgeWithValue<CfLongId,Float>> edges, ComputeFunction.Callback<CfLongId,Svdpp.SvdppValue,Float,FloatMatrixMessage> cb) The function for computing a new vertex value or sending messages to the next superstep.
void	preSuperstep(int superstep, ComputeFunction.Aggregators aggregators) Prepare for computation.
protected void	updateValue(org.jblas.FloatMatrix user, org.jblas.FloatMatrix item, float error, float gamma, float lambda)

Methods inherited from class java.lang.Object

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

Methods inherited from interface io.kgraph.pregel.ComputeFunction

init, masterCompute, postSuperstep

Constructor Detail

UserComputation

public UserComputation()

Method Detail

updateValue

protected void updateValue(org.jblas.FloatMatrix user,
                           org.jblas.FloatMatrix item,
                           float error,
                           float gamma,
                           float lambda)

preSuperstep

public void preSuperstep(int superstep,
                         ComputeFunction.Aggregators aggregators)

Description copied from interface: ComputeFunction

Prepare for computation. This method is executed exactly once prior to compute() being called for any of the vertices in the partition.

Specified by:

preSuperstep in interface ComputeFunction<CfLongId,Svdpp.SvdppValue,Float,FloatMatrixMessage>

Parameters:

superstep - the superstep

aggregators - the aggregators

compute

public void compute(int superstep,
                    VertexWithValue<CfLongId,Svdpp.SvdppValue> vertex,
                    Iterable<FloatMatrixMessage> messages,
                    Iterable<EdgeWithValue<CfLongId,Float>> edges,
                    ComputeFunction.Callback<CfLongId,Svdpp.SvdppValue,Float,FloatMatrixMessage> cb)

Description copied from interface: ComputeFunction

The function for computing a new vertex value or sending messages to the next superstep.

Specified by:

compute in interface ComputeFunction<CfLongId,Svdpp.SvdppValue,Float,FloatMatrixMessage>

Parameters:

superstep - the count of the current superstep

vertex - the current vertex with its value

messages - a Map of the source vertex and the message sent from the previous superstep

edges - the adjacent edges with their values

cb - a callback for setting a new vertex value or sending messages to the next superstep