com.intel.analytics.bigdl.example

textclassification

package textclassification

Visibility
  1. Public
  2. All

Type Members

  1. case class TextClassificationParams(baseDir: String = "./", maxSequenceLength: Int = 1000, maxWordsNum: Int = 20000, trainingSplit: Double = 0.8, batchSize: Int = 128, embeddingDim: Int = 100, partitionNum: Int = 4) extends Product with Serializable

  2. class TextClassifier extends AnyRef

    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.

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

Value Members

  1. object SimpleTokenizer

  2. object TextClassifier

Ungrouped