|
@ -1,7 +1,15 @@ |
|
|
|
|
|
from . import data |
|
|
from . import dataset |
|
|
from . import dataset |
|
|
from . import label |
|
|
from . import label |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def get_data_class(project_type: str): |
|
|
|
|
|
if project_type in ['DocumentClassification', 'SequenceLabeling', 'Seq2seq']: |
|
|
|
|
|
return data.TextData |
|
|
|
|
|
else: |
|
|
|
|
|
return data.FileData |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def get_dataset_class(format: str): |
|
|
def get_dataset_class(format: str): |
|
|
if format == 'csv': |
|
|
if format == 'csv': |
|
|
return dataset.CsvDataset |
|
|
return dataset.CsvDataset |
|
|