com.intel.analytics.bigdl.example.utils

TextClassificationParams

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

baseDir

The root directory which containing the training and embedding data

maxSequenceLength

number of the tokens

maxWordsNum

maximum word to be included

trainingSplit

percentage of the training data

batchSize

size of the mini-batch

embeddingDim

size of the embedding vector

learningRate

learning rate

Linear Supertypes
Product, Equals, AbstractTextClassificationParams, Serializable, Serializable, AnyRef, Any
Ordering
  1. Alphabetic
  2. By inheritance
Inherited
  1. TextClassificationParams
  2. Product
  3. Equals
  4. AbstractTextClassificationParams
  5. Serializable
  6. Serializable
  7. AnyRef
  8. Any
  1. Hide All
  2. Show all
Learn more about member selection
Visibility
  1. Public
  2. All

Instance Constructors

  1. new 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)

    baseDir

    The root directory which containing the training and embedding data

    maxSequenceLength

    number of the tokens

    maxWordsNum

    maximum word to be included

    trainingSplit

    percentage of the training data

    batchSize

    size of the mini-batch

    embeddingDim

    size of the embedding vector

    learningRate

    learning rate

Value Members

  1. final def !=(arg0: AnyRef): Boolean

    Definition Classes
    AnyRef
  2. final def !=(arg0: Any): Boolean

    Definition Classes
    Any
  3. final def ##(): Int

    Definition Classes
    AnyRef → Any
  4. final def ==(arg0: AnyRef): Boolean

    Definition Classes
    AnyRef
  5. final def ==(arg0: Any): Boolean

    Definition Classes
    Any
  6. final def asInstanceOf[T0]: T0

    Definition Classes
    Any
  7. val baseDir: String

    The root directory which containing the training and embedding data

    The root directory which containing the training and embedding data

    Definition Classes
    TextClassificationParamsAbstractTextClassificationParams
  8. val batchSize: Int

    size of the mini-batch

    size of the mini-batch

    Definition Classes
    TextClassificationParamsAbstractTextClassificationParams
  9. def clone(): AnyRef

    Attributes
    protected[java.lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  10. val embeddingDim: Int

    size of the embedding vector

    size of the embedding vector

    Definition Classes
    TextClassificationParamsAbstractTextClassificationParams
  11. final def eq(arg0: AnyRef): Boolean

    Definition Classes
    AnyRef
  12. def finalize(): Unit

    Attributes
    protected[java.lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( classOf[java.lang.Throwable] )
  13. final def getClass(): Class[_]

    Definition Classes
    AnyRef → Any
  14. final def isInstanceOf[T0]: Boolean

    Definition Classes
    Any
  15. val learningRate: Double

    learning rate

  16. val maxSequenceLength: Int

    number of the tokens

    number of the tokens

    Definition Classes
    TextClassificationParamsAbstractTextClassificationParams
  17. val maxWordsNum: Int

    maximum word to be included

    maximum word to be included

    Definition Classes
    TextClassificationParamsAbstractTextClassificationParams
  18. final def ne(arg0: AnyRef): Boolean

    Definition Classes
    AnyRef
  19. final def notify(): Unit

    Definition Classes
    AnyRef
  20. final def notifyAll(): Unit

    Definition Classes
    AnyRef
  21. val partitionNum: Int

  22. final def synchronized[T0](arg0: ⇒ T0): T0

    Definition Classes
    AnyRef
  23. val trainingSplit: Double

    percentage of the training data

    percentage of the training data

    Definition Classes
    TextClassificationParamsAbstractTextClassificationParams
  24. final def wait(): Unit

    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  25. final def wait(arg0: Long, arg1: Int): Unit

    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  26. final def wait(arg0: Long): Unit

    Definition Classes
    AnyRef
    Annotations
    @throws( ... )

Inherited from Product

Inherited from Equals

Inherited from Serializable

Inherited from Serializable

Inherited from AnyRef

Inherited from Any

Ungrouped