mrseanryan / csfy

Train a simple text classifier and predict labels - supports ONNX output for performance, language-neutral
https://medium.com/p/9fd7003000fc
MIT License
0 stars 0 forks source link

Add train/test split option #7

Open KONRADS098 opened 3 days ago

KONRADS098 commented 3 days ago

Add train/test split option to allow to open up the possibility for evaluation

Add a new command that splits the data into the provided split (train_size, test_size), add shuffle and random_state options for shuffling the data set before splitting.