Commit b9f8cb46 authored by Gaëtan Caillaut's avatar Gaëtan Caillaut
Browse files

mélange le dataframe

parent fe69ba87
......@@ -586,7 +586,7 @@ def split_neighbors(df, outdir):
for th, x in themes.items():
subdf = df[df["token"].isin(x)]
outpath = str(Path(outdir, f"{th}_minibert.csv"))
subdf.to_csv(outpath, index=False)
subdf.sample(frac=1).to_csv(outpath, index=False)
def compute_neighbors_for_sncf(args):
for md in os.listdir(args.path):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment