Machine learning system for classifying Python code snippets into functional categories using TF-IDF vectorization and Random Forest classification.
- TF-IDF code tokenization
- Random Forest classification
- Hyperparameter tuning with GridSearch
- Confusion matrix visualization
- Prediction pipeline
- Python
- scikit-learn
- pandas
- numpy
pip install -r requirements.txt python main.py