2023-04-11 14:00:54 +00:00
|
|
|
import datasets
|
|
|
|
|
|
|
|
|
|
test_ds = datasets.load_dataset("dataset/rotor37_data.py", split="test")
|
|
|
|
|
test_ds = test_ds.with_format("torch")
|
|
|
|
|
print(test_ds)
|
|
|
|
|
|
2023-04-11 15:32:30 +00:00
|
|
|
train_ds = datasets.load_dataset("dataset/rotor37_data.py", split="train")
|
|
|
|
|
train_ds = train_ds.with_format("torch")
|
|
|
|
|
print(train_ds)
|
|
|
|
|
|
2023-04-11 14:00:54 +00:00
|
|
|
print("yay")
|
