README

Overview

This repository contains the code and documentation for the internship project "Finding Vulnerabilities Using AI" conducted at DRDO's Scientific Analysis Group. The project focuses on leveraging AI and machine learning to detect vulnerabilities in web applications, specifically targeting Cross-Site Scripting (XSS) and SQL Injection (SQLi).

Contents

  • Report
    • Detailed internship report documenting the project phases, methodologies, and results.
  • Data Collection & Preprocessing
    • Scripts for merging and preprocessing datasets from Kaggle.
  • Model Training & Testing
    • Jupyter notebooks for training and evaluating Logistic Regression, Random Forest, and XGBoost models.
  • Prediction
    • Scripts to predict vulnerabilities in the provided web application codebase.

Acknowledgements

Grateful thanks to Shri Amrendra Kumar for his mentorship and Aditya sir for technical guidance. Special thanks to my parents and friends for their support.