Class/Object

org.apache.spark.ml.recommendation

TfIdfKNNModel

Related Docs: object TfIdfKNNModel | package recommendation

Permalink

class TfIdfKNNModel extends Model[TfIdfKNNModel] with _TfIdfKNN with KNNRecommender with MLWritable

Model fitted by TfIdfKNN.

Linear Supertypes
MLWritable, KNNRecommender, _TfIdfKNN, KNNParams, HasPredictionCol, Model[TfIdfKNNModel], Transformer, PipelineStage, Logging, Params, Serializable, Serializable, Identifiable, AnyRef, Any
Ordering
  1. Alphabetic
  2. By Inheritance
Inherited
  1. TfIdfKNNModel
  2. MLWritable
  3. KNNRecommender
  4. _TfIdfKNN
  5. KNNParams
  6. HasPredictionCol
  7. Model
  8. Transformer
  9. PipelineStage
  10. Logging
  11. Params
  12. Serializable
  13. Serializable
  14. Identifiable
  15. AnyRef
  16. Any
  1. Hide All
  2. Show All
Visibility
  1. Public
  2. All

Instance Constructors

  1. new TfIdfKNNModel(uid: String, idfDf: DataFrame, userDf: DataFrame)

    Permalink

    uid

    The UID

    idfDf

    The data frame with IDF weightings for all items

    userDf

    The data frame of all users and items the model was fitted on, and their TF-IDF weightings

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 $[T](param: Param[T]): T

    Permalink
    Attributes
    protected
    Definition Classes
    Params
  4. final def ==(arg0: Any): Boolean

    Permalink
    Definition Classes
    AnyRef → Any
  5. final def asInstanceOf[T0]: T0

    Permalink
    Definition Classes
    Any
  6. final def clear(param: Param[_]): TfIdfKNNModel.this.type

    Permalink
    Definition Classes
    Params
  7. def clone(): AnyRef

    Permalink
    Attributes
    protected[java.lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  8. def copy(extra: ParamMap): TfIdfKNNModel

    Permalink
    Definition Classes
    TfIdfKNNModel → Model → Transformer → PipelineStage → Params
  9. def copyValues[T <: Params](to: T, extra: ParamMap): T

    Permalink
    Attributes
    protected
    Definition Classes
    Params
  10. final def defaultCopy[T <: Params](extra: ParamMap): T

    Permalink
    Attributes
    protected
    Definition Classes
    Params
  11. final def eq(arg0: AnyRef): Boolean

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

    Permalink
    Definition Classes
    AnyRef → Any
  13. def explainParam(param: Param[_]): String

    Permalink
    Definition Classes
    Params
  14. def explainParams(): String

    Permalink
    Definition Classes
    Params
  15. final def extractParamMap(): ParamMap

    Permalink
    Definition Classes
    Params
  16. final def extractParamMap(extra: ParamMap): ParamMap

    Permalink
    Definition Classes
    Params
  17. final val filterUserItems: BooleanParam

    Permalink

    Whether the items of a user should be filtered from the recommendations for the user Default: false

    Whether the items of a user should be filtered from the recommendations for the user Default: false

    Definition Classes
    KNNParams
  18. def finalize(): Unit

    Permalink
    Attributes
    protected[java.lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( classOf[java.lang.Throwable] )
  19. final def get[T](param: Param[T]): Option[T]

    Permalink
    Definition Classes
    Params
  20. final def getClass(): Class[_]

    Permalink
    Definition Classes
    AnyRef → Any
  21. final def getDefault[T](param: Param[T]): Option[T]

    Permalink
    Definition Classes
    Params
  22. def getFilterUserItems: Boolean

    Permalink

    Definition Classes
    KNNParams
  23. def getItemCol: String

    Permalink

    Definition Classes
    KNNParams
  24. def getK: Int

    Permalink

    Definition Classes
    KNNParams
  25. final def getOrDefault[T](param: Param[T]): T

    Permalink
    Definition Classes
    Params
  26. def getParam(paramName: String): Param[Any]

    Permalink
    Definition Classes
    Params
  27. final def getPredictionCol: String

    Permalink
    Definition Classes
    HasPredictionCol
  28. def getUserCol: String

    Permalink

    Definition Classes
    KNNParams
  29. final def hasDefault[T](param: Param[T]): Boolean

    Permalink
    Definition Classes
    Params
  30. def hasParam(paramName: String): Boolean

    Permalink
    Definition Classes
    Params
  31. def hasParent: Boolean

    Permalink
    Definition Classes
    Model
  32. def hashCode(): Int

    Permalink
    Definition Classes
    AnyRef → Any
  33. val idfDf: DataFrame

    Permalink

    The data frame with IDF weightings for all items

  34. def initializeLogIfNecessary(isInterpreter: Boolean, silent: Boolean): Boolean

    Permalink
    Attributes
    protected
    Definition Classes
    Logging
  35. def initializeLogIfNecessary(isInterpreter: Boolean): Unit

    Permalink
    Attributes
    protected
    Definition Classes
    Logging
  36. final def isDefined(param: Param[_]): Boolean

    Permalink
    Definition Classes
    Params
  37. final def isInstanceOf[T0]: Boolean

    Permalink
    Definition Classes
    Any
  38. final def isSet(param: Param[_]): Boolean

    Permalink
    Definition Classes
    Params
  39. def isTraceEnabled(): Boolean

    Permalink
    Attributes
    protected
    Definition Classes
    Logging
  40. final val itemCol: Param[String]

    Permalink

    The name of the item id column of integers from 0 to number of items in training dataset Default: "itemid"

    The name of the item id column of integers from 0 to number of items in training dataset Default: "itemid"

    Definition Classes
    KNNParams
  41. final val k: IntParam

    Permalink

    The number of nearest neighbours Default: 150

    The number of nearest neighbours Default: 150

    Definition Classes
    KNNParams
  42. def log: Logger

    Permalink
    Attributes
    protected
    Definition Classes
    Logging
  43. def logDebug(msg: ⇒ String, throwable: Throwable): Unit

    Permalink
    Attributes
    protected
    Definition Classes
    Logging
  44. def logDebug(msg: ⇒ String): Unit

    Permalink
    Attributes
    protected
    Definition Classes
    Logging
  45. def logError(msg: ⇒ String, throwable: Throwable): Unit

    Permalink
    Attributes
    protected
    Definition Classes
    Logging
  46. def logError(msg: ⇒ String): Unit

    Permalink
    Attributes
    protected
    Definition Classes
    Logging
  47. def logInfo(msg: ⇒ String, throwable: Throwable): Unit

    Permalink
    Attributes
    protected
    Definition Classes
    Logging
  48. def logInfo(msg: ⇒ String): Unit

    Permalink
    Attributes
    protected
    Definition Classes
    Logging
  49. def logName: String

    Permalink
    Attributes
    protected
    Definition Classes
    Logging
  50. def logTrace(msg: ⇒ String, throwable: Throwable): Unit

    Permalink
    Attributes
    protected
    Definition Classes
    Logging
  51. def logTrace(msg: ⇒ String): Unit

    Permalink
    Attributes
    protected
    Definition Classes
    Logging
  52. def logWarning(msg: ⇒ String, throwable: Throwable): Unit

    Permalink
    Attributes
    protected
    Definition Classes
    Logging
  53. def logWarning(msg: ⇒ String): Unit

    Permalink
    Attributes
    protected
    Definition Classes
    Logging
  54. final def ne(arg0: AnyRef): Boolean

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

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

    Permalink
    Definition Classes
    AnyRef
  57. lazy val params: Array[Param[_]]

    Permalink
    Definition Classes
    Params
  58. var parent: Estimator[TfIdfKNNModel]

    Permalink
    Definition Classes
    Model
  59. final val predictionCol: Param[String]

    Permalink
    Definition Classes
    HasPredictionCol
  60. def recommendForUserSubset(queryDataset: Dataset[_], numItems: Int): DataFrame

    Permalink

    Returns top numItems items recommended for each user id in the input data set

    Returns top numItems items recommended for each user id in the input data set

    queryDataset

    The dataset containing a column of user ids. The column name must match userCol

    numItems

    The maximum number of recommendations for each user

    returns

    A dataframe of (userCol: Int, recommendations), where recommendations are stored as an array of (score: Float, itemCol: Int) rows. Or if exploded a dataframe of (userCol: Int, score: Float, itemCol: Int) rows.

  61. def recommendFromNearestUsers(queryDf: DataFrame, nnDf: DataFrame, numItems: Int): DataFrame

    Permalink
    Definition Classes
    KNNRecommender
  62. def save(path: String): Unit

    Permalink
    Definition Classes
    MLWritable
    Annotations
    @Since( "1.6.0" ) @throws( ... )
  63. final def set(paramPair: ParamPair[_]): TfIdfKNNModel.this.type

    Permalink
    Attributes
    protected
    Definition Classes
    Params
  64. final def set(param: String, value: Any): TfIdfKNNModel.this.type

    Permalink
    Attributes
    protected
    Definition Classes
    Params
  65. final def set[T](param: Param[T], value: T): TfIdfKNNModel.this.type

    Permalink
    Definition Classes
    Params
  66. final def setDefault(paramPairs: ParamPair[_]*): TfIdfKNNModel.this.type

    Permalink
    Attributes
    protected
    Definition Classes
    Params
  67. final def setDefault[T](param: Param[T], value: T): TfIdfKNNModel.this.type

    Permalink
    Attributes
    protected
    Definition Classes
    Params
  68. def setFilterUserItems(value: Boolean): TfIdfKNNModel.this.type

    Permalink

    Definition Classes
    KNNRecommender
  69. def setK(value: Int): TfIdfKNNModel.this.type

    Permalink

    Definition Classes
    KNNRecommender
  70. def setParent(parent: Estimator[TfIdfKNNModel]): TfIdfKNNModel

    Permalink
    Definition Classes
    Model
  71. final def synchronized[T0](arg0: ⇒ T0): T0

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

    Permalink
    Definition Classes
    Identifiable → AnyRef → Any
  73. def toTfIdf(idfDf: DataFrame): DataFrame

    Permalink
    Definition Classes
    _TfIdfKNN
  74. def transform(dataset: Dataset[_]): DataFrame

    Permalink
    Definition Classes
    TfIdfKNNModel → Transformer
  75. def transform(dataset: Dataset[_], paramMap: ParamMap): DataFrame

    Permalink
    Definition Classes
    Transformer
    Annotations
    @Since( "2.0.0" )
  76. def transform(dataset: Dataset[_], firstParamPair: ParamPair[_], otherParamPairs: ParamPair[_]*): DataFrame

    Permalink
    Definition Classes
    Transformer
    Annotations
    @Since( "2.0.0" ) @varargs()
  77. def transformSchema(schema: StructType): StructType

    Permalink
    Definition Classes
    TfIdfKNNModel → PipelineStage
  78. def transformSchema(schema: StructType, logging: Boolean): StructType

    Permalink
    Attributes
    protected
    Definition Classes
    PipelineStage
    Annotations
    @DeveloperApi()
  79. val uid: String

    Permalink

    The UID

    The UID

    Definition Classes
    TfIdfKNNModel → Identifiable
  80. final val userCol: Param[String]

    Permalink

    The name of the user id column of integers Default: "userid"

    The name of the user id column of integers Default: "userid"

    Definition Classes
    KNNParams
  81. val userDf: DataFrame

    Permalink

    The data frame of all users and items the model was fitted on, and their TF-IDF weightings

  82. final def wait(): Unit

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

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

    Permalink
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  85. def write: MLWriter

    Permalink
    Definition Classes
    TfIdfKNNModel → MLWritable

Inherited from MLWritable

Inherited from KNNRecommender

Inherited from _TfIdfKNN

Inherited from KNNParams

Inherited from HasPredictionCol

Inherited from Model[TfIdfKNNModel]

Inherited from Transformer

Inherited from PipelineStage

Inherited from Logging

Inherited from Params

Inherited from Serializable

Inherited from Serializable

Inherited from Identifiable

Inherited from AnyRef

Inherited from Any

getParam

param

setParam

Ungrouped