Need private packages and team management tools?Check out npm Teams »

tvt-split

1.0.3 • Public • Published

NPM

GitHub package version devDependencies Status dependencies Status

GH Downloads GitHub commit activity the past year GitHub contributors Github search hit counter

Build Status codecov.io Code Coverage tested with jest Known Vulnerabilities

GitHub contributions welcome Commitizen friendly

GitHub top language GitHub language count GitHub code size in bytes

BCH compliance Codacy Badge

Split your dataset into training, validation and test datasets.

This was largely inspired by train-test-split.

Install

npm install tvt-split
#or with Yarn 
yarn add tvt-split

Documentation

Table of Contents

trainValidationTestSplit

Split data into a training, validation and test set.

Parameters

  • data Array Data
  • train number Size of the training set (in percentage if train < 1)
  • validation number Size of the validation set (in percentage if train < 1)
  • seed number Seed of the Pseudo-Random Number Generation. (optional, default -1.1)
  • indices boolean Indicates whether to return the indices of the provided array or the actual values (optional, default false)

Returns [Array, Array, Array] Split data

Example usage

const tvts = require('tvt-split');
 
const arr = ['one', 'two', 'three', 'four', 'five', 'six', 'seven', 'eight', 'nine', 'ten'];
const [train, validation, test] = tvts(arr, 0.8, .1, 1234);
console.log(train, validation, test);
// [ 'three', 'one', 'seven', 'eight', 'five', 'four', 'nine', 'six' ] [ 'two' ] [ 'ten' ]
 
const [trainIndices, validationIndices, testIndices] = tvts(arr, 0.8, .1, 1234, true);
console.log(trainIndices, validationIndices, testIndices);
// [ 2, 0, 6, 7, 4, 3, 8, 5, ] [ 1 ] [ 9 ]

Install

npm i tvt-split

DownloadsWeekly Downloads

16

Version

1.0.3

License

MIT

Unpacked Size

29.4 kB

Total Files

31

Last publish

Collaborators

  • avatar