Machine learning model

Question

Can someone help me with the below requirement which is:

I want to build a model that should understand a piece of code and predict the programming language based on the code contents.

Any lead would be highly appreciated.

eg:

private int input = 0;
 
public int getter() {
   return input;
}
public void setter(int in) {
  input = in;
}

Output - Java

Thanks

Answer 1

Maybe use some pretrained language model to run on the test data. Test data can be some statements and the language they belong to and you can proceed as a classification task.

Answer 2

You can use N-grams to understand which keyword belongs to which language. You can check it out: ngram resource

You can find source codes by web scraping through GitHub. So, you can create a source code and programming language dataset by yourself. If you want to go deeper, there are books about this topic. programming language detection book