• Home
• com.microsoft.ml.spark
• mmlspark_2.11
• 0.18.1
• source code
• SchemaConstants.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.

com.microsoft.ml.spark.core.schema.SchemaConstants.scala Maven / Gradle / Ivy

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

package com.microsoft.ml.spark.core.schema

/** Contains constants used by modules for schema. */
object SchemaConstants {

  val ScoreColumnKind           = "ScoreColumnKind"
  val ScoreValueKind            = "ScoreValueKind"

  val TrueLabelsColumn          = "true_labels"
  val ScoredLabelsColumn        = "scored_labels"
  val ScoresColumn              = "scores"
  val ScoredProbabilitiesColumn = "scored_probabilities"

  val ScoreModelPrefix          = "score_model"
  val MMLTag                    = "mml"      // MML metadata tag
  val MLlibTag                  = "ml_attr"  // MLlib metadata tag, see org.apache.spark.ml.attribute.AttributeKeys

  /** The following tags are used in Metadata representation of categorical data
    * do not change them or use them directly
    * (see org.apache.spark.ml.attribute.AttributeKeys for the first three)
    */
  val Ordinal                   = "ord"        // common tag for both MLlib and MML
  val MLlibTypeTag              = "type"       // MLlib tag for the attribute types
  val ValuesString              = "vals"       // common tag for both MLlib and MML
  val ValuesInt                 = "vals_int"
  val ValuesLong                = "vals_long"
  val ValuesDouble              = "vals_double"
  val ValuesBool                = "vals_bool"
  val HasNullLevels             = "null_exists"

  // Score value kinds, or types of ML:
  val ClassificationKind        = "Classification"
  val RegressionKind            = "Regression"

  // Spark native column names
  val SparkPredictionColumn     = "prediction"
  val SparkRawPredictionColumn  = "rawPrediction"
  val SparkProbabilityColumn    = "probability"

}