This Python program classifies tumour cell datapoints as benign or malignant using a Support Vector Machine (SVM) algorithm. The program extracts data from a cell_samples.csv
, preprocesses it, trains a SVM model, and then uses the trained model to classify new data.
- Python 3.x
- pandas
- scikit-learn
- Clone the repository to your local machine:
git clone https://github.com/yourusername/SVM-cancer-cell-classification.git
- Install the required libraries using pip:
pip install pandas scikit-learn
- Run the
main.py
file to run the program.
You can customize the model architecture, hyperparameters, and other settings by modifying the code in the main.py
file. You can also update the preprocessing steps in the main.py
file according to your specific requirements.
If you would like to contribute to this project, please follow the standard GitHub fork and pull request workflow.
- Article: https://medium.com/@opemipo404/classifying-tumour-cells-with-support-vector-machines-838e6396a8da
- Tutorial: https://youtu.be/7sz4WpkUIIs
This project is licensed under the MIT License.
If you have any questions, suggestions, or issues, please feel fre to contact me at opethepope@gmail.com.