diff --git a/data/datasets/cdae_data_pipeline.py b/data/datasets/cdae_data_pipeline.py index 6261f5d..81210cb 100644 --- a/data/datasets/cdae_data_pipeline.py +++ b/data/datasets/cdae_data_pipeline.py @@ -18,10 +18,9 @@ def split(self, df): ''' logger.info(f'start random user split...') - item_list = df.columns[1:].astype(int) train_data, valid_data, test_data = {}, {}, {} - for i, row in tqdm(df.iterrows(), total=df.shape[0]): + for _, row in tqdm(df.iterrows(), total=df.shape[0]): # train+valid+test mask user_id = int(row['user_id'])