التقسيم والمجموعات والزمن والتسرب

import numpy as np
groups = np.array(["a", "a", "b", "b", "c", "c"])
train = np.flatnonzero(groups != "c")
test = np.flatnonzero(groups == "c")
print(train.tolist(), test.tolist())
print(sorted(set(groups[train]) & set(groups[test])))

[0, 1, 2, 3] [4, 5]
[]

تختلف الفهارس ولا تظهر مجموعة بالطرفين، لكنه يفحص التعريف لا كل استقلال.

import numpy as np
prediction_time = np.array([1, 2, 3, 4, 5, 6])
label_available = prediction_time + 1
cutoff = 5
train = label_available < cutoff
test = prediction_time >= cutoff
print("training times", prediction_time[train].tolist())
print("test times", prediction_time[test].tolist())

training times [1, 2, 3]
test times [5, 6]

يستبعد الوقت أربعة لأن هدفه يتوفر بخمسة وفق الحد الصارم، لا فجوة عالمية.