Options
All
  • Public
  • Public/Protected
  • All
Menu

Interface TrainTestSplitOptions

Hierarchy

  • TrainTestSplitOptions

Index

Properties

seed?: string

seed for random number generator

shuffle?: boolean

If True, shuffle the data before splitting it. Recommended for most ML applications

stratify?: any[] | Matrix

array of categories, e.g. class labels, which is used to stratifiy the holdout train/test split

trainFraction?: number

number greater than 0, smaller than 1, indicating the fraction of data used for the training settypically 0.8