This project aims to predict cancer diagnosis (malignant or benign) using machine learning techniques. The dataset used in this project is cancer.csv
.
The dataset cancer.csv
contains information regarding various features related to tumors and their diagnosis. The target variable is diagnosis
, where 1 represents malignant tumors and 0 represents benign tumors.
To run this project, ensure you have Python installed along with the following libraries:
- pandas
- scikit-learn
- TensorFlow
You can install these dependencies using pip:
pip install pandas scikit-learn tensorflow
- Clone this repository to your local machine.
- Navigate to the project directory.
- Run the Python script to execute the model.
python cancer_prediction.py
- The dataset is loaded using pandas.
- Features (x) and target (y) variables are separated.
- The dataset is split into training and testing sets using train_test_split from scikit-learn.
- A Sequential model is created using TensorFlow's Keras API.
- The model architecture consists of Dense layers with sigmoid activation.
- The model is compiled with Adam optimizer and binary crossentropy loss.
- Training is performed for 2000 epochs.
Gonçalo Alves
This project is licensed under the MIT License.