MCPcopy Create free account
hub / github.com/OpenBMB/BMTools / create_split

Function create_split

bmtools/tools/database/utils/data.py:16–27  ·  view source on GitHub ↗

Split data into two parts. Data is in the form of a tuple of lists.

(data, split_size)

Source from the content-addressed store, hash-verified

14
15
16def create_split(data, split_size):
17 """
18 Split data into two parts. Data is in the form of a tuple of lists.
19 """
20 inputs, outputs = data
21 assert len(inputs) == len(outputs)
22 indices = random.sample(range(len(inputs)), split_size)
23 inputs1 = [inputs[i] for i in indices]
24 outputs1 = [outputs[i] for i in indices]
25 inputs2 = [inputs[i] for i in range(len(inputs)) if i not in indices]
26 outputs2 = [outputs[i] for i in range(len(inputs)) if i not in indices]
27 return (inputs1, outputs1), (inputs2, outputs2)

Callers

nothing calls this directly

Calls

no outgoing calls

Tested by

no test coverage detected