Package

com.intel.analytics.bigdl.example

udfpredictor

Permalink

package udfpredictor

Visibility
  1. Public
  2. All

Type Members

  1. case class TextClassificationUDFParams(baseDir: String = "./", maxSequenceLength: Int = 1000, maxWordsNum: Int = 20000, trainingSplit: Double = 0.8, batchSize: Int = 128, embeddingDim: Int = 100, partitionNum: Int = 4, modelPath: Option[String] = None, checkpoint: Option[String] = None, testDir: String = "./") extends AbstractTextClassificationParams with Product with Serializable

    Permalink

    Text classification udf parameters

Value Members

  1. object DataframePredictor

    Permalink
  2. object Utils

    Permalink

Ungrouped