com.intel.analytics.bigdl.example

utils

package utils

Visibility
  1. Public
  2. All

Type Members

  1. abstract class AbstractTextClassificationParams extends Serializable

  2. case class TextClassificationParams(baseDir: String = "./", maxSequenceLength: Int = 500, maxWordsNum: Int = 5000, trainingSplit: Double = 0.8, batchSize: Int = 128, embeddingDim: Int = 200, learningRate: Double = 0.01, partitionNum: Int = 4) extends AbstractTextClassificationParams with Product with Serializable

  3. class TextClassifier extends Serializable

    This example use a (pre-trained GloVe embedding) to convert word to vector, and uses it to train a text classification model on the 20 Newsgroup dataset with 20 different categories.

  4. case class WordMeta(count: Int, index: Int) extends Product with Serializable

Value Members

  1. object SimpleTokenizer

Ungrouped