Pre-defined task improvements #662
Labels
discojs
Related to Disco.js
feature
New feature or request
web client
Related to the browser environment
Image tasks support choosing a train-validation split in which a percentage of the training set is set apart as a validation set (
discojs-core/src/dataset/data_loader/image_loader.ts
). In some cases however, e.g., when using imbalanced or small datasets, this can result in some classes not being included in the validation set at all.Unfortunately, it seems that tensorflow.js doesn't offer any function for stratified splits.
The text was updated successfully, but these errors were encountered: