public final class StratifiedSplit extends GeneratedMessageV3 implements StratifiedSplitOrBuilder
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.
Protobuf type google.cloud.aiplatform.v1beta1.StratifiedSplit
Static Fields
public static final int KEY_FIELD_NUMBER
Field Value
public static final int TEST_FRACTION_FIELD_NUMBER
Field Value
public static final int TRAINING_FRACTION_FIELD_NUMBER
Field Value
public static final int VALIDATION_FRACTION_FIELD_NUMBER
Field Value
Static Methods
public static StratifiedSplit getDefaultInstance()
Returns
public static final Descriptors.Descriptor getDescriptor()
Returns
public static StratifiedSplit.Builder newBuilder()
Returns
public static StratifiedSplit.Builder newBuilder(StratifiedSplit prototype)
Parameter
Returns
public static StratifiedSplit parseDelimitedFrom(InputStream input)
Parameter
Returns
Exceptions
public static StratifiedSplit parseDelimitedFrom(InputStream input, ExtensionRegistryLite extensionRegistry)
Parameters
Returns
Exceptions
public static StratifiedSplit parseFrom(byte[] data)
Parameter
Name | Description |
data | byte[]
|
Returns
Exceptions
public static StratifiedSplit parseFrom(byte[] data, ExtensionRegistryLite extensionRegistry)
Parameters
Returns
Exceptions
public static StratifiedSplit parseFrom(ByteString data)
Parameter
Returns
Exceptions
public static StratifiedSplit parseFrom(ByteString data, ExtensionRegistryLite extensionRegistry)
Parameters
Returns
Exceptions
public static StratifiedSplit parseFrom(CodedInputStream input)
Parameter
Returns
Exceptions
public static StratifiedSplit parseFrom(CodedInputStream input, ExtensionRegistryLite extensionRegistry)
Parameters
Returns
Exceptions
public static StratifiedSplit parseFrom(InputStream input)
Parameter
Returns
Exceptions
public static StratifiedSplit parseFrom(InputStream input, ExtensionRegistryLite extensionRegistry)
Parameters
Returns
Exceptions
public static StratifiedSplit parseFrom(ByteBuffer data)
Parameter
Returns
Exceptions
public static StratifiedSplit parseFrom(ByteBuffer data, ExtensionRegistryLite extensionRegistry)
Parameters
Returns
Exceptions
public static Parser<StratifiedSplit> parser()
Returns
Methods
public boolean equals(Object obj)
Parameter
Returns
Overrides
public StratifiedSplit getDefaultInstanceForType()
Returns
Required. The key is a name of one of the Dataset's data columns.
The key provided must be for a categorical column.
string key = 4 [(.google.api.field_behavior) = REQUIRED];
Returns
public ByteString getKeyBytes()
Required. The key is a name of one of the Dataset's data columns.
The key provided must be for a categorical column.
string key = 4 [(.google.api.field_behavior) = REQUIRED];
Returns
public Parser<StratifiedSplit> getParserForType()
Returns
Overrides
public int getSerializedSize()
Returns
Overrides
public double getTestFraction()
The fraction of the input data that is to be used to evaluate the Model.
double test_fraction = 3;
Returns
Type | Description |
double | The testFraction.
|
public double getTrainingFraction()
The fraction of the input data that is to be used to train the Model.
double training_fraction = 1;
Returns
Type | Description |
double | The trainingFraction.
|
public final UnknownFieldSet getUnknownFields()
Returns
Overrides
public double getValidationFraction()
The fraction of the input data that is to be used to validate the Model.
double validation_fraction = 2;
Returns
Type | Description |
double | The validationFraction.
|
Returns
Overrides
protected GeneratedMessageV3.FieldAccessorTable internalGetFieldAccessorTable()
Returns
Overrides
public final boolean isInitialized()
Returns
Overrides
public StratifiedSplit.Builder newBuilderForType()
Returns
protected StratifiedSplit.Builder newBuilderForType(GeneratedMessageV3.BuilderParent parent)
Parameter
Returns
Overrides
protected Object newInstance(GeneratedMessageV3.UnusedPrivateParameter unused)
Parameter
Returns
Overrides
public StratifiedSplit.Builder toBuilder()
Returns
public void writeTo(CodedOutputStream output)
Parameter
Overrides
Exceptions