No Description

Alexander Levin 3c1219b23c added: regexed v6 data 3 weeks ago
.dvc bd896b3eeb copy from github; 4 months ago
.ipynb_checkpoints bd896b3eeb copy from github; 4 months ago
analysis 9de45f715a moved to "analysis" 3 weeks ago
data
data_mini
deprecated 9075bc827f moved to "deprecated" folder 3 weeks ago
graph dc835f45bb upd: github sync; 3 months ago
models
models_scripts a8ab4d2cb5 fix: evaluation process 3 weeks ago
open_data
parser a264c3e091 added "parser" folder 3 weeks ago
.gitattributes f23a7f09e6 git cloned from https://github.com/whatevernevermindbro/source_code_classification 5 months ago
.gitignore fba9f1690b added: vscode, ipynb_checkpoints 3 weeks ago
Comments vs commented code.ipynb f23a7f09e6 git cloned from https://github.com/whatevernevermindbro/source_code_classification 5 months ago
Makefile bd896b3eeb copy from github; 4 months ago
README.md ae877d8024 added: conventions 1 month ago
data.dvc 3c1219b23c added: regexed v6 data 3 weeks ago
data_mini.dvc d0d261341a upd: data unified with sep=',' and CODE_COLUMN = 'code_block' 3 weeks ago
metrics.csv 4bb2cbaa47 experiment: regex_v6 3 weeks ago
models.dvc dc835f45bb upd: github sync; 3 months ago
nl2ml.pptx dc835f45bb upd: github sync; 3 months ago
open_data.dvc bd896b3eeb copy from github; 4 months ago
params.yml 4bb2cbaa47 experiment: regex_v6 3 weeks ago
requirements.txt dcb234b9ff init: requirements file 1 month ago

Data Pipeline

Legend
DVC Managed File
Git Managed File
Metric
Stage File
External File

README.md

Source Code Classification

This is a repo of the Natural Language to Machine Learning (NL2ML) project of the Laboratory of Methods for Big Data Analysis at Higher School of Economics (HSE LAMBDA).

The project's official repo is stored on GitLab (HSE LAMBDA repository) - https://gitlab.com/lambda-hse/nl2ml \ The project's full description is stored on Notion - https://www.notion.so/NL2ML-Corpus-1ed964c08eb049b383c73b9728c3a231 \ The project's experiments are stored on DAGsHub - https://dagshub.com/levin/source_code_classification

Project Goals

Short-Term Goal

To build a model classifying a source code chunk and to specify where the detected class is exactly in the chunk (tag segmentation).

Long-Term Goal

To build a model generating code by getting a short raw english task in as an input.

Repository Description

This repository contains instruments which the project's team has been using to label source code chunks with Knowledge Graph vertices and to train models to recognize these vertices in future. By the Knowledge Graph vertices we mean an elementary part of ML-pipeline. The current latest version of the Knowledge Graph contains the following high-level vertices: ['import', 'data_import', 'data_export', 'preprocessing', 'visualization', 'model', 'deep_learning_model', 'train' 'predict'].

Data Download

To download the project data and models:

  1. Clone this repository
  2. Install DVC from https://dvc.org/doc/install
  3. Do dvc pull data or dvc pull data. Note: if you are failing on dvc pull [folder_to_pull], try dvc pull [folder_to_pull] --jobs 1

Contents:

The instruments which we have been using to reach the project goals are: notebooks parsing from Kaggle API and Github API, data preparation, regex-labellig, training models, validation models, model weights/coefficients analysis, errors analysis, synonyms analysis.

nl2ml_notebook_parser.py - a script for parsing Kaggle notebooks and process them to JSON/CSV/Pandas.

bert_distances.ipynb - a notebook with BERT expiremints concerning sense of distance between BERT embeddings where input tokens were tokenized source code chunks.

bert_classifier.ipynb - a notebook with preprocessing and training BERT-pipeline.

regex.ipynb - a notebook with creating labels for code chunks with regex

logreg_classifier.ipynb - a notebook with training logistic regression model on the regex labels with tf-idf and analyzing the outputs

Comments vs commented code.ipynb - a notebook with a model distinguishing NL-comments from commented source code

github_dataset.ipynb - a notebook with opening github_dataset

predict_tag.ipynb - a notebook with predicting class label (tag) with any model

svm_classifier.ipynb - a notebook with training SVM (replaced by _svmtrain.py) and analyzing SVM outputs

svm_train.py - a script for training SVM model

Conventions:

  • Input CSV: encoding='utf-8', sep=',' and CODE_COLUMN has to be == 'code_block' in all input CSVs
  • Knowledge Graphs: GRAPH_DIR has to be in the following format: './graph/graph_v{}.txt'.format(GRAPH_VER)