Initializes the DataLoader with a file path or an existing DataFrame.
Parameters: |
-
file
(str , default:
None
)
–
Path to the CSV file to load.
-
dataframe
(DataFrame , default:
None
)
–
|
Raises: |
-
ValueError
–
If neither file nor dataframe is provided.
|
Source code in LabeLMaker/Evaluate/data_loader.py
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26 | def __init__(
self, file: Optional[str] = None, dataframe: Optional[pd.DataFrame] = None
) -> None:
"""
Initializes the DataLoader with a file path or an existing DataFrame.
Parameters:
file (str, optional): Path to the CSV file to load.
dataframe (pd.DataFrame, optional): An existing DataFrame.
Raises:
ValueError: If neither file nor dataframe is provided.
"""
if file is not None:
self.df = self.load_csv_file(file)
elif dataframe is not None:
self.df = dataframe
else:
raise ValueError("Either 'file' or 'dataframe' must be provided.")
|