minor bug fixes (#207)

* fixed path arg for create_dataset_csvs()

* updated assign_check() to remove user warning
This commit is contained in:
Daniel Kleine
2024-06-12 13:27:56 +02:00
committed by GitHub
parent b2ff989174
commit e5c3c5ce99
2 changed files with 5 additions and 14 deletions

View File

@@ -117,7 +117,7 @@ def random_split(df, train_frac, validation_frac):
return train_df, validation_df, test_df
def create_dataset_csvs(data_file_path):
def create_dataset_csvs(new_file_path):
df = pd.read_csv(new_file_path, sep="\t", header=None, names=["Label", "Text"])
# Create balanced dataset