Generating embeddings for Python and Java about code2seq HOT 5 CLOSED

Hi Avra,
Thank you for your interest in this work! Sorry again for the delayed response.

Yes, in order to use code2vec for python, you will have to train the model on a python dataset.
Have you seen this section in the README? https://github.com/tech-srl/code2vec#extending-to-other-languages

Uri

from code2seq.

@urialon. astminer team helped me producing training, testing and validation python.c2v data. So, how we should proceed next please to train code2vec model? Then once we train the model, can we feed the same data to produce embeddings as we have 20k python files that we split into train, test and validate before feeding them to astminer tool. Once we train code2vec, we would like to feed the same python code to produce embeddings, so what do you think please?

from code2seq.

I am not sure how your python.c2v look like, but try to continue running the preprocess.sh script starting from this line: https://github.com/tech-srl/code2seq/blob/master/preprocess.sh#L54 (and adapt all paths according to your files).

from code2seq.

@urialon.

Thank you. So I will train your model on 150k python dataset. How to please save the model to use it later on another python dataset to generate embeddings? Does the preprocessor.sh does it automatically and save the model please?

We would like too once we train the model on 150k python dataset you specified to use to generate later on one embedding vector for each python file we have in our own dataset, can we do that please? We don't want to generate method name but one embedding that is representative of a file. We would like to do the same for our 20k python files.

from code2seq.

Hi @Avra2 ,

preprocess.sh just preprocesses the data, it does not even train the model.
However, train.sh trains and saves the checkpoints. See:
https://github.com/tech-srl/code2vec#training-a-model-from-scratch

from code2seq.