Function write_file

data/amazon23_data_process.py:370–381

(path, data)

    print(f"[Split] train={len(train_data)}, valid={len(valid_data)}, test={len(test_data)}")

    def write_file(path, data):
        with open(path, "w") as f:
            f.write("user_id:token\titem_id_list:token_seq\titem_id:token\n")
            for it in data:
                user_original = it[0]
                uid = user2index[user_original]

                hist = [str(x) for x in it[3]]
                target = str(it[4])

                hist = hist[-50:]  # cap history length = 50
                f.write(f"{uid}\t{' '.join(hist)}\t{target}\n")

    write_file(os.path.join(out_dir, f"{args.dataset}.train.inter"), train_data)
    write_file(os.path.join(out_dir, f"{args.dataset}.valid.inter"), valid_data)