All the installations, processing and training steps are same as the original implementation of code2vec.
This repository explores a way to work with this dataset and modifies the java extractor code(already there in code2vec implementation) to label data with its respective class complexity.
Demo
Results:
Although it's still work in progress but so far we have 70% accuracy, along with ~60% for precision, recall and F1 score respectively.