trait GPTGenerationParams extends Params
Linear Supertypes
Known Subclasses
Ordering
- Grouped
- Alphabetic
- By Inheritance
Inherited
- GPTGenerationParams
- Params
- Serializable
- Serializable
- Identifiable
- AnyRef
- Any
- Hide All
- Show All
Visibility
- Public
- All
Abstract Value Members
Concrete Value Members
-
final
def
!=(arg0: Any): Boolean
- Definition Classes
- AnyRef → Any
-
final
def
##(): Int
- Definition Classes
- AnyRef → Any
-
final
def
$[T](param: Param[T]): T
- Attributes
- protected
- Definition Classes
- Params
-
final
def
==(arg0: Any): Boolean
- Definition Classes
- AnyRef → Any
-
final
def
asInstanceOf[T0]: T0
- Definition Classes
- Any
-
final
def
clear(param: Param[_]): GPTGenerationParams.this.type
- Definition Classes
- Params
-
def
clone(): AnyRef
- Attributes
- protected[lang]
- Definition Classes
- AnyRef
- Annotations
- @throws( ... ) @native()
-
def
copyValues[T <: Params](to: T, extra: ParamMap): T
- Attributes
- protected
- Definition Classes
- Params
-
final
def
defaultCopy[T <: Params](extra: ParamMap): T
- Attributes
- protected
- Definition Classes
- Params
-
val
doSample: BooleanParam
Whether or not to use sampling, use greedy decoding otherwise (Default:
false
) -
final
def
eq(arg0: AnyRef): Boolean
- Definition Classes
- AnyRef
-
def
equals(arg0: Any): Boolean
- Definition Classes
- AnyRef → Any
-
def
explainParam(param: Param[_]): String
- Definition Classes
- Params
-
def
explainParams(): String
- Definition Classes
- Params
-
final
def
extractParamMap(): ParamMap
- Definition Classes
- Params
-
final
def
extractParamMap(extra: ParamMap): ParamMap
- Definition Classes
- Params
-
def
finalize(): Unit
- Attributes
- protected[lang]
- Definition Classes
- AnyRef
- Annotations
- @throws( classOf[java.lang.Throwable] )
-
final
def
get[T](param: Param[T]): Option[T]
- Definition Classes
- Params
-
final
def
getClass(): Class[_]
- Definition Classes
- AnyRef → Any
- Annotations
- @native()
-
final
def
getDefault[T](param: Param[T]): Option[T]
- Definition Classes
- Params
- def getDoSample: Boolean
- def getIgnoreTokenIds: Array[Int]
- def getMaxContextLength: Int
- def getMaxNewTokens: Int
- def getNoRepeatNgramSize: Int
-
final
def
getOrDefault[T](param: Param[T]): T
- Definition Classes
- Params
-
def
getParam(paramName: String): Param[Any]
- Definition Classes
- Params
- def getRandomSeed: Option[Int]
- def getTopK: Int
-
final
def
hasDefault[T](param: Param[T]): Boolean
- Definition Classes
- Params
-
def
hasParam(paramName: String): Boolean
- Definition Classes
- Params
-
def
hashCode(): Int
- Definition Classes
- AnyRef → Any
- Annotations
- @native()
-
val
ignoreTokenIds: IntArrayParam
A list of token ids which are ignored in the decoder's output (Default:
Array()
) -
final
def
isDefined(param: Param[_]): Boolean
- Definition Classes
- Params
-
final
def
isInstanceOf[T0]: Boolean
- Definition Classes
- Any
-
final
def
isSet(param: Param[_]): Boolean
- Definition Classes
- Params
-
val
maxContextLength: IntParam
Maximum length of context text.
Maximum length of context text. (Default:
1024
) -
val
maxNewTokens: IntParam
Maximum number of new tokens to be generated (Default: 30)
-
final
def
ne(arg0: AnyRef): Boolean
- Definition Classes
- AnyRef
-
val
noRepeatNgramSize: IntParam
If set to int >
0
, all ngrams of that size can only occur once (Default:0
) -
final
def
notify(): Unit
- Definition Classes
- AnyRef
- Annotations
- @native()
-
final
def
notifyAll(): Unit
- Definition Classes
- AnyRef
- Annotations
- @native()
-
lazy val
params: Array[Param[_]]
- Definition Classes
- Params
-
val
randomSeed: Option[Int]
Optional Random seed for the model.
Optional Random seed for the model. Needs to be of type
Long
. -
final
def
set(paramPair: ParamPair[_]): GPTGenerationParams.this.type
- Attributes
- protected
- Definition Classes
- Params
-
final
def
set(param: String, value: Any): GPTGenerationParams.this.type
- Attributes
- protected
- Definition Classes
- Params
-
final
def
set[T](param: Param[T], value: T): GPTGenerationParams.this.type
- Definition Classes
- Params
-
final
def
setDefault(paramPairs: ParamPair[_]*): GPTGenerationParams.this.type
- Attributes
- protected
- Definition Classes
- Params
-
final
def
setDefault[T](param: Param[T], value: T): GPTGenerationParams.this.type
- Attributes
- protected[org.apache.spark.ml]
- Definition Classes
- Params
- def setDoSample(value: Boolean): GPTGenerationParams.this.type
- def setIgnoreTokenIds(tokenIds: Array[Int]): GPTGenerationParams.this.type
- def setMaxContextLength(value: Int): GPTGenerationParams.this.type
- def setMaxNewTokens(value: Int): GPTGenerationParams.this.type
- def setNoRepeatNgramSize(value: Int): GPTGenerationParams.this.type
- def setRandomSeed(value: Int): GPTGenerationParams.this.type
- def setTopK(value: Int): GPTGenerationParams.this.type
-
final
def
synchronized[T0](arg0: ⇒ T0): T0
- Definition Classes
- AnyRef
-
def
toString(): String
- Definition Classes
- Identifiable → AnyRef → Any
-
val
topK: IntParam
The number of highest probability vocabulary tokens to consider
-
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
- @throws( ... ) @native()