CodeNetPy Code Classification

Machine learning system for classifying Python code snippets into functional categories using TF-IDF vectorization and Random Forest classification.

Features

• TF-IDF code tokenization
• Random Forest classification
• Hyperparameter tuning with GridSearch
• Confusion matrix visualization
• Prediction pipeline

Tech Stack

• Python
• scikit-learn
• pandas
• numpy

Run

pip install -r requirements.txt python main.py