-
Notifications
You must be signed in to change notification settings - Fork 1.2k
Closed
Labels
component: trainingRelates to the SageMaker Training PlatformRelates to the SageMaker Training Platformtype: feature request
Description
Describe the feature you'd like
An input argument to fit() that automatically splits the training data into train & validation sets.
How would this feature be used? Please describe.
Input float between 0 and 1 to fraction of the training data to be used as validation data.
i.e. instead of
model.fit(inputs={'train': s3_input("s3://.../train"), 'validation': s3_input("s3://.../valid")}) it would be something like:
model.fit(inputs={'train': s3_input("s3://.../train"), 'validation_split': 0.2}) and 20% of the training data would be held away as a validation set.
Describe alternatives you've considered
See the implementation in keras: https://keras.rstudio.com/reference/fit.html
GISTech01
Metadata
Metadata
Assignees
Labels
component: trainingRelates to the SageMaker Training PlatformRelates to the SageMaker Training Platformtype: feature request