StratifiedSplit class final

Assigns input data to the training, validation, and test sets so that the distribution of values found in the categorical column (as specified by the key field) is mirrored within each split. The fraction values determine the relative sizes of the splits.

For example, if the specified column has three values, with 50% of the rows having value "A", 25% value "B", and 25% value "C", and the split fractions are specified as 80/10/10, then the training set will constitute 80% of the training data, with about 50% of the training set rows having the value "A" for the specified column, about 25% having the value "B", and about 25% having the value "C".

Only the top 500 occurring values are used; any values not in the top 500 values are randomly assigned to a split. If less than three rows contain a specific value, those rows are randomly assigned.

Supported only for tabular Datasets.

Inheritance
  • Object
  • ProtoMessage
  • StratifiedSplit

Constructors

StratifiedSplit({double trainingFraction = 0, double validationFraction = 0, double testFraction = 0, required String key})
StratifiedSplit.fromJson(Object? j)
factory

Properties

hashCode → int
The hash code for this object.
no setterinherited
key → String
Required. The key is a name of one of the Dataset's data columns. The key provided must be for a categorical column.
final
qualifiedName → String
The fully qualified name of this message, i.e., google.protobuf.Duration or google.rpc.ErrorInfo.
finalinherited
runtimeType → Type
A representation of the runtime type of the object.
no setterinherited
testFraction → double
The fraction of the input data that is to be used to evaluate the Model.
final
trainingFraction → double
The fraction of the input data that is to be used to train the Model.
final
validationFraction → double
The fraction of the input data that is to be used to validate the Model.
final

Methods

noSuchMethod(Invocation invocation) → dynamic
Invoked when a nonexistent method or property is accessed.
inherited
toJson() → Object
override
toString() → String
A string representation of this object.
override

Operators

operator ==(Object other) → bool
The equality operator.
inherited

Constants

fullyQualifiedName → const String