12 lines
291 B
Python
12 lines
291 B
Python
|
import datasets
|
||
|
|
|
||
|
|
train_ds = datasets.load_dataset("dataset/rotor37_data.py", split="train")
|
||
|
|
train_ds = train_ds.with_format("torch")
|
||
|
|
print(train_ds)
|
||
|
|
|
||
|
|
test_ds = datasets.load_dataset("dataset/rotor37_data.py", split="test")
|
||
|
|
test_ds = test_ds.with_format("torch")
|
||
|
|
print(test_ds)
|
||
|
|
|
||
|
|
print("yay")
|