14 lines
317 B
Python
14 lines
317 B
Python
|
import pandas as pd
|
||
|
|
import os
|
||
|
|
|
||
|
|
from .version import Version
|
||
|
|
from .base import BaseDataset
|
||
|
|
from .info import DatasetInfo
|
||
|
|
from . import constants as C
|
||
|
|
|
||
|
|
|
||
|
|
def get_dataset(split="train"):
|
||
|
|
meta = pd.read_csv("../data/meta.csv")
|
||
|
|
info = DatasetInfo.load("../data/info.yaml")
|
||
|
|
return BaseDataset(info, meta)[split]
|