(data, train_rate, valid_rate, test_rate)
| 44 | |
| 45 | |
| 46 | def division(data, train_rate, valid_rate, test_rate): |
| 47 | data_length = len(data) |
| 48 | train_division_index = int(data_length * train_rate) |
| 49 | valid_division_index = int(data_length * (train_rate + valid_rate)) |
| 50 | test_division_index = int(data_length * (1 - test_rate)) |
| 51 | train_data = data[:train_division_index, :] |
| 52 | valid_data = data[train_division_index:valid_division_index, :] |
| 53 | test_data = data[test_division_index:, :] |
| 54 | return train_data, valid_data, test_data |
| 55 | |
| 56 | |
| 57 | def set_seed(seed, flag): |
nothing calls this directly
no outgoing calls
no test coverage detected