levin
/
source_code_classification
mirror of https://github.com/whatevernevermindbro/source_code_classification


  
1

	
2

	
3

	
4

	
5

	
6

	
7

	
            graph = {'imports':['import', 'from']
 ,'load_data':['load', 'csv', 'glob', 'pickle', 'read', 'download', 'txt']
 ,'preprocessing':['preprocess', 'plot', 'heatmap', 'figure', 'encode', 'fill', 'merge', 'vectorize', 'generation', 'augmentation', 'stemming', 'lemmatization', 'scale', 'normalize', 'crop', 'convert', 'resize']
 ,'model':['LinearRegression', 'RandomForest', 'Ridge', 'Lasso', 'SGD', 'LogisticRegression', 'SVM', 'SVC', 'Layer', 'xgboost', 'LGBM', 'CatBoost']
 ,'train': ['train', 'fit', 'epoch', 'loss', 'learn', 'optimizer']
 ,'predict':['predict', 'error', 'evaluation', 'MSE', 'Test']
}