• Home
• com.microsoft.azure
• synapseml-vw_2.12
• 1.0.0
• source code
• HasNumBits.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.azure.synapse.ml.vw.HasNumBits.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.azure.synapse.ml.vw

import org.apache.spark.ml.param.{IntParam, Params}

/** Controls hashing parameters such us number of bits (numbits) and how to handle collisions.
  */
trait HasNumBits extends Params {
  val numBits = new IntParam(this, "numBits", "Number of bits used to mask")
  setDefault(numBits -> 30)

  def getNumBits: Int = $(numBits)

  def setNumBits(value: Int): this.type = {
    if (value < 1 || value > 30)
      throw new IllegalArgumentException("Number of bits must be between 1 and 30 bits")
    set(numBits, value)
  }

  /**
    * @return the bitmask used to constrain the hash feature indices.
    */
  protected def getMask: Int = (1 << getNumBits) - 1
}