Which Rows Sit Out?

10 pts · 10 coins

Machine Learning in CodeEasy

A teacher wants to hold back some exam papers as a fair test. Each paper has an ID number.

Task: write test_ids(ids, test_size, seed).

ids is a list of ID numbers.
Split it with train_test_split(ids, test_size=test_size, random_state=seed).
Return the IDs that went into the test part, sorted from smallest to largest.

ids = [101, 102, 103, 104, 105, 106, 107, 108, 109, 110]

test_size = 0.3

seed = 0

returns [103, 105, 109]

ids = list(range(1, 21))

test_size = 0.25

seed = 42

returns [1, 2, 9, 16, 18]

solution.py

1from sklearn.model_selection import train_test_split
2
3
4def test_ids(ids, test_size, seed):
5    pass
6