csinva / imodelsX

Scikit-learn friendly library to interpret, and prompt-engineer text datasets using large language models.
https://csinva.io/imodelsX/
MIT License
149 stars 25 forks source link

train_test_split test_size and shuffle parameters #12

Closed rafa-rod closed 2 months ago

rafa-rod commented 2 months ago

The size of test size will be chosen by user. In case of time series, shuffle=False is required, so it is import the user change it.

csinva commented 2 months ago

Thanks, nice addition!