1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
|
#!/usr/bin/env python
import openturns as ot
N = 10
splitters = [ot.LeaveOneOutSplitter(N)]
splitters.append(ot.KFoldSplitter(N, 3))
splitters.append(ot.KFoldSplitter(N, 5))
kfold = ot.KFoldSplitter(N, 5)
kfold.setRandomize(True)
splitters.append(kfold)
splitters.append(ot.KFoldSplitter(N, N))
for splitter in splitters:
print(f"splitter={splitter}")
size = 0
for indices1, indices2 in splitter:
print(f"indices={indices1}, {indices2}")
assert len(indices1) + len(indices2) == N, "not enough indices"
assert indices1.check(N), "dup or too large indices1"
assert indices2.check(N), "dup or too large indices2"
for i1 in indices1:
assert i1 not in indices2, "indices1 in indices2"
for i2 in indices2:
assert i2 not in indices1, "indices2 in indices1"
size += 1
assert size == splitter.getSize(), "wrong size"
# try selection
indices = [8, 0]
sample = ot.Normal().getSample(N)
sample_i = sample[indices]
print(f"sample={sample} indices={indices} sample_i={sample_i}")
assert len(sample_i) == len(indices), "wrong selection len"
assert sample_i[0] == sample[indices[0]], "wrong first item in selection"
|
