• Home
• com.microsoft.ml.spark
• mmlspark_2.11
• 0.18.1
• source code
• Ranker.scala

×

Project download

Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $

You can buy this project and download/modify it how often you want.

org.apache.spark.ml.Ranker.scala Maven / Gradle / Ivy

// Copyright (C) Microsoft Corporation. All rights reserved.
// Licensed under the MIT License. See LICENSE in project root for information.

package org.apache.spark.ml

// Note: a bit strange to have the mmlspark import here, but it works
import com.microsoft.ml.spark.core.contracts.HasGroupCol

/**
  * Ranker base class
  *
  * @tparam FeaturesType Type of input features.  E.g., org.apache.spark.mllib.linalg.Vector
  * @tparam Learner      Concrete Estimator type
  * @tparam M            Concrete Model type
  */
abstract class Ranker[FeaturesType, Learner <: Ranker[FeaturesType, Learner, M], M <: RankerModel[FeaturesType, M]]
  extends Predictor[FeaturesType, Learner, M] with PredictorParams with HasGroupCol

/**
  * Model produced by a `Ranker`.
  *
  * @tparam FeaturesType Type of input features.  E.g., org.apache.spark.mllib.linalg.Vector
  * @tparam M            Concrete Model type.
  */
abstract class RankerModel[FeaturesType, M <: RankerModel[FeaturesType, M]]
  extends PredictionModel[FeaturesType, M] with PredictorParams