case class TrainParams(dataFolder: String = "./", modelSnapshot: Option[String] = None, stateSnapshot: Option[String] = None, checkpoint: Option[String] = None, batchSize: Int = 20, learningRate: Double = 0.01, learningRateDecay: Double = 0.001, hiddenSize: Int = 200, vocabSize: Int = 10000, nEpochs: Int = 4, numLayers: Int = 2, numSteps: Int = 20, overWriteCheckpoint: Boolean = false, keepProb: Float = 2.0f, withTransformerModel: Boolean = false, optimizerVersion: Option[String] = None) extends Product with Serializable
Linear Supertypes
Ordering
- Alphabetic
- By Inheritance
Inherited
- TrainParams
- Serializable
- Serializable
- Product
- Equals
- AnyRef
- Any
- Hide All
- Show All
Visibility
- Public
- All
Instance Constructors
- new TrainParams(dataFolder: String = "./", modelSnapshot: Option[String] = None, stateSnapshot: Option[String] = None, checkpoint: Option[String] = None, batchSize: Int = 20, learningRate: Double = 0.01, learningRateDecay: Double = 0.001, hiddenSize: Int = 200, vocabSize: Int = 10000, nEpochs: Int = 4, numLayers: Int = 2, numSteps: Int = 20, overWriteCheckpoint: Boolean = false, keepProb: Float = 2.0f, withTransformerModel: Boolean = false, optimizerVersion: Option[String] = None)
Value Members
-
final
def
!=(arg0: Any): Boolean
- Definition Classes
- AnyRef → Any
-
final
def
##(): Int
- Definition Classes
- AnyRef → Any
-
final
def
==(arg0: Any): Boolean
- Definition Classes
- AnyRef → Any
-
final
def
asInstanceOf[T0]: T0
- Definition Classes
- Any
- val batchSize: Int
- val checkpoint: Option[String]
-
def
clone(): AnyRef
- Attributes
- protected[java.lang]
- Definition Classes
- AnyRef
- Annotations
- @native() @throws( ... )
- val dataFolder: String
-
final
def
eq(arg0: AnyRef): Boolean
- Definition Classes
- AnyRef
-
def
finalize(): Unit
- Attributes
- protected[java.lang]
- Definition Classes
- AnyRef
- Annotations
- @throws( classOf[java.lang.Throwable] )
-
final
def
getClass(): Class[_]
- Definition Classes
- AnyRef → Any
- Annotations
- @native()
- val hiddenSize: Int
-
final
def
isInstanceOf[T0]: Boolean
- Definition Classes
- Any
- val keepProb: Float
- val learningRate: Double
- val learningRateDecay: Double
- val modelSnapshot: Option[String]
- val nEpochs: Int
-
final
def
ne(arg0: AnyRef): Boolean
- Definition Classes
- AnyRef
-
final
def
notify(): Unit
- Definition Classes
- AnyRef
- Annotations
- @native()
-
final
def
notifyAll(): Unit
- Definition Classes
- AnyRef
- Annotations
- @native()
- val numLayers: Int
- val numSteps: Int
- val optimizerVersion: Option[String]
- val overWriteCheckpoint: Boolean
- val stateSnapshot: Option[String]
-
final
def
synchronized[T0](arg0: ⇒ T0): T0
- Definition Classes
- AnyRef
- val vocabSize: Int
-
final
def
wait(): Unit
- Definition Classes
- AnyRef
- Annotations
- @throws( ... )
-
final
def
wait(arg0: Long, arg1: Int): Unit
- Definition Classes
- AnyRef
- Annotations
- @throws( ... )
-
final
def
wait(arg0: Long): Unit
- Definition Classes
- AnyRef
- Annotations
- @native() @throws( ... )
- val withTransformerModel: Boolean