Class

com.tencent.angel.ml.factorizationmachines

FMLearner

Related Doc: package factorizationmachines

Permalink

class FMLearner extends MLLearner

Learner of Factorization machines

Linear Supertypes
MLLearner, AnyRef, Any
Ordering
  1. Alphabetic
  2. By Inheritance
Inherited
  1. FMLearner
  2. MLLearner
  3. AnyRef
  4. Any
  1. Hide All
  2. Show All
Visibility
  1. Public
  2. All

Instance Constructors

  1. new FMLearner(ctx: TaskContext, minP: Double, maxP: Double, feaUsed: Array[Int])

    Permalink

Value Members

  1. final def !=(arg0: Any): Boolean

    Permalink
    Definition Classes
    AnyRef → Any
  2. final def ##(): Int

    Permalink
    Definition Classes
    AnyRef → Any
  3. final def ==(arg0: Any): Boolean

    Permalink
    Definition Classes
    AnyRef → Any
  4. val LOG: Log

    Permalink
  5. final def asInstanceOf[T0]: T0

    Permalink
    Definition Classes
    Any
  6. def clone(): AnyRef

    Permalink
    Attributes
    protected[java.lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  7. val conf: Configuration

    Permalink
    Definition Classes
    MLLearner
  8. val ctx: TaskContext

    Permalink
    Definition Classes
    FMLearnerMLLearner
  9. def derviationMultipler(y: Double, pre: Double): Double

    Permalink

    \frac{\partial loss}{\partial x} = dm * \frac{\partial y}{\partial x}

    \frac{\partial loss}{\partial x} = dm * \frac{\partial y}{\partial x}

    returns

    : dm value

  10. val epochNum: Int

    Permalink
  11. final def eq(arg0: AnyRef): Boolean

    Permalink
    Definition Classes
    AnyRef
  12. def equals(arg0: Any): Boolean

    Permalink
    Definition Classes
    AnyRef → Any
  13. def evaluate(dataBlock: DataBlock[LabeledData], w0: Double, w: DenseDoubleVector, v: HashMap[Int, DenseDoubleVector]): Double

    Permalink

    Evaluate the objective value For regression: loss(y,\hat y) = (y - \hat y)2 For classification: loss(y,\hat y) = -\ln (\delta(y, \hat y))), in which \delta(x) = \frac{1}{1+e{-x}}

  14. val feaNum: Int

    Permalink
  15. val feaUsed: Array[Int]

    Permalink
  16. val feaUsedN: Int

    Permalink
  17. def finalize(): Unit

    Permalink
    Attributes
    protected[java.lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( classOf[java.lang.Throwable] )
  18. val fmmodel: FMModel

    Permalink
  19. final def getClass(): Class[_]

    Permalink
    Definition Classes
    AnyRef → Any
  20. val globalMetrics: GlobalMetrics

    Permalink
    Definition Classes
    MLLearner
  21. def hashCode(): Int

    Permalink
    Definition Classes
    AnyRef → Any
  22. def initModels(): Unit

    Permalink

    Initialize with random values

  23. final def isInstanceOf[T0]: Boolean

    Permalink
    Definition Classes
    Any
  24. val learnType: String

    Permalink
  25. val lr: Double

    Permalink
  26. val maxP: Double

    Permalink
  27. val minP: Double

    Permalink
  28. final def ne(arg0: AnyRef): Boolean

    Permalink
    Definition Classes
    AnyRef
  29. final def notify(): Unit

    Permalink
    Definition Classes
    AnyRef
  30. final def notifyAll(): Unit

    Permalink
    Definition Classes
    AnyRef
  31. def oneIteration(dataBlock: DataBlock[LabeledData]): (DenseDoubleVector, DenseDoubleVector, HashMap[Int, DenseDoubleVector])

    Permalink

    One iteration to train Factorization Machines

  32. def predict(x: SparseDoubleSortedVector, y: Double, w0: Double, w: DenseDoubleVector, v: HashMap[Int, DenseDoubleVector]): Double

    Permalink

    Predict an instance

  33. val rank: Int

    Permalink
  34. val reg0: Double

    Permalink
  35. val reg1: Double

    Permalink
  36. val reg2: Double

    Permalink
  37. final def synchronized[T0](arg0: ⇒ T0): T0

    Permalink
    Definition Classes
    AnyRef
  38. def toString(): String

    Permalink
    Definition Classes
    AnyRef → Any
  39. def train(trainData: DataBlock[LabeledData], vali: DataBlock[LabeledData]): MLModel

    Permalink

    Train a Factorization machines Model

    Train a Factorization machines Model

    trainData

    : input train data storage

    vali

    : validate data storage

    returns

    : a learned model

    Definition Classes
    FMLearnerMLLearner
  40. def updateV(x: SparseDoubleSortedVector, dm: Double, v: HashMap[Int, DenseDoubleVector]): Unit

    Permalink

    Update v mat

  41. val vIndexs: RowIndex

    Permalink
  42. val vStddev: Double

    Permalink
  43. final def wait(): Unit

    Permalink
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  44. final def wait(arg0: Long, arg1: Int): Unit

    Permalink
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  45. final def wait(arg0: Long): Unit

    Permalink
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )

Inherited from MLLearner

Inherited from AnyRef

Inherited from Any

Ungrouped