WxxShirley / KDD2024ProCom

Codes and data for KDD 2024 Research Track paper "ProCom: A Few-shot Targeted Community Detection Algorithm"
https://arxiv.org/abs/2408.07369
MIT License
4 stars 1 forks source link

Model Transferability Test #1

Closed Jillian555 closed 1 month ago

Jillian555 commented 1 month ago

Table 5 shows the transferability test of the method. When loading the pre-trained model, how to align the feature dimensions of the source domain and the target domain, for example, the feature dimension of the pretrain dataset is 10, and the feature dimension of the target dataset is 13.

WxxShirley commented 1 month ago

Sorry for late reply.

To ensure transferability, we maintain a consistent input dimension by uniformly setting the input features based on their structural patterns (lines 35-57 in code/data.py). This enables that each node from any dataset has a consistent input dimension of 5. In our main experiments, even with attributed graphs like Twitter and Facebook, we utilize their structural features instead of the original features.

If you have any further questions, feel free to discuss!