main.py: error: unrecognized arguments: --agg concat

[XueniLuo]

Hi, I run the recommended code
python main.py --dataset ali --gnn ngcf --dim 64 --lr 0.0001 --batch_size 1024 --gpu_id 0 --context_hops 3 --agg concat --ns mixgcf --K 1 --n_negs 64

And received the issue:
main.py: error: unrecognized arguments: --agg concat;

I checked the file /utils/parser.py, and found the --agg parameter is not declared in that file.

Is there anything I missed?

Thanks for your time and have a nice day.

[mushding]

According to modules/LightGCN.py line 157-166

    def pooling(self, embeddings):
        # [-1, n_hops, channel]
        if self.pool == 'mean':
            return embeddings.mean(dim=1)
        elif self.pool == 'sum':
            return embeddings.sum(dim=1)
        elif self.pool == 'concat':
            return embeddings.view(embeddings.shape[0], -1)
        else:  # final
            return embeddings[:, -1, :]

And utils/parser.py line 33

parser.add_argument("--pool", type=str, default='concat', help="[concat, mean, sum, final]")

the "--agg" argument should be "--pool"

python main.py --dataset ali --gnn ngcf --dim 64 --lr 0.0001 --batch_size 1024 --gpu_id 0 --context_hops 3 --pool concat --ns mixgcf --K 1 --n_negs 64

fix that typo, issue should be solved.

[XueniLuo]

According to modules/LightGCN.py line 157-166

    def pooling(self, embeddings):
        # [-1, n_hops, channel]
        if self.pool == 'mean':
            return embeddings.mean(dim=1)
        elif self.pool == 'sum':
            return embeddings.sum(dim=1)
        elif self.pool == 'concat':
            return embeddings.view(embeddings.shape[0], -1)
        else:  # final
            return embeddings[:, -1, :]

And utils/parser.py line 33

parser.add_argument("--pool", type=str, default='concat', help="[concat, mean, sum, final]")

the "--agg" argument should be "--pool"

python main.py --dataset ali --gnn ngcf --dim 64 --lr 0.0001 --batch_size 1024 --gpu_id 0 --context_hops 3 --pool concat --ns mixgcf --K 1 --n_negs 64

fix that typo, issue should be solved.

Thanks a lot, that completely solves my problem.
Have a nice day~