Skip to main content

Python Tip: Precise Control over Scikit-Learn Train-Test Splits

·1 min

Post Image

Despite using train_test_split for some time now, I just realized you can specify an integer to the “test_size” argument of this function to specify the exact number of samples you want to get as your test size.

Those little things you learn when you actually read the documentation ðŸ«