diff options
Diffstat (limited to 'megapixels/commands/msc/summarize.py')
| -rw-r--r-- | megapixels/commands/msc/summarize.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/megapixels/commands/msc/summarize.py b/megapixels/commands/msc/summarize.py index d5d251db..045e3b69 100644 --- a/megapixels/commands/msc/summarize.py +++ b/megapixels/commands/msc/summarize.py @@ -29,7 +29,7 @@ def cli(ctx, opt_fp_in, opt_fp_out): log = Logger.getLogger() - dataset_names = ['helen', 'megaface', 'adience', 'pipa', 'lfpw', 'duke_mtmc', 'brainwash', 'msceleb', 'uccs'] + dataset_names = ['helen', 'megaface', 'adience', 'pipa', 'lfpw', 'brainwash', 'msceleb', 'duke_mtmc', 'uccs'] df = pd.DataFrame() fp_out = opt_fp_out.replace('.csv', '_citations.csv') @@ -37,10 +37,11 @@ def cli(ctx, opt_fp_in, opt_fp_out): fp_csv = join(opt_fp_in, f'{dataset_name}.csv') _df = pd.read_csv(fp_csv) _df = _df[_df.lat != 0] + _df.drop('id', axis=1, inplace=True) print(dataset_name, len(_df)) df = df.append(_df, ignore_index=True) - df.to_csv(opt_fp_out, index=False) + df.to_csv(fp_out, index=False) # create country summary fp_out = opt_fp_out.replace('.csv', '_countries.csv') |