torch_frame.utils.generate_random_split

generate_random_split(length: int, seed: int, train_ratio: float = 0.8, val_ratio: float = 0.1, include_test: bool = True) → ndarray

生成指定长度的随机分割分配列表。 元素为0、1或2，分别代表训练、验证和测试。请注意，此函数依赖于numpy的洗牌在不同版本之间是一致的这一事实，这在历史上一直是如此。