Model for other task. about code2vec HOT 2 OPEN

Hi @XuRuiAngel,

By default, code2vec is only able to interpret single methods. However, some other studies have tried to overcome this limitation. For example, Compton et al. have used various aggregation methods on the method embeddings to create class-level embeddings. You could experiment yourself with some aggregation methods to combine the various java files and use the resulting embeddings to train a classifier on your task of classifying design patterns.

from code2vec.

Hi @XuRuiAngel ,
Thank you for your interest in our work.

As @daveymathijssen said, code2vec is currently working on a single function at a time.
We have other recent work such as PolyCoder https://github.com/VHellendoorn/Code-LMs
and a CodeBERT model that we fine-tuned on Java, that maybe you can fine-tune to your task: https://huggingface.co/neulab/codebert-java

Let me know if you have any more questions.
Best,
Uri

from code2vec.