Skip to content

Latest commit

Β 

History

History
65 lines (42 loc) Β· 1.67 KB

File metadata and controls

65 lines (42 loc) Β· 1.67 KB

🧠 ANN Classification

This project demonstrates a simple Artificial Neural Network (ANN) for classification tasks using Keras and TensorFlow. It is designed to classify structured data into multiple or binary classes depending on the dataset used.


πŸ“š Overview

This neural network is built with fully connected layers using Keras's Sequential API. It includes data preprocessing, model training, evaluation, and visualization.


πŸ› οΈ Project Structure

ANN-Classification-/ β”‚ β”œβ”€β”€ ann_classification.ipynb # Jupyter Notebook with full pipeline β”œβ”€β”€ dataset.csv # Input dataset (placeholder name) β”œβ”€β”€ model.h5 # Saved ANN model β”œβ”€β”€ requirements.txt # Python dependencies └── README.md # Project documentation


πŸš€ Features

  • Build and train an Artificial Neural Network from scratch
  • Preprocess tabular data using normalization/encoding
  • Evaluate model performance using accuracy and confusion matrix
  • Save and load trained models

πŸ“Š Dataset

Replace this with the name of the dataset used (e.g., Iris, Banknote Authentication, etc.)

  • Contains numerical and/or categorical features
  • Target column represents class label(s)

🧠 ANN Architecture

  • Input Layer: Depends on the number of features
  • Hidden Layers: Dense layers with ReLU activation
  • Output Layer:
    • Sigmoid (for binary classification)
    • Softmax (for multi-class classification)
  • Loss Function:
    • binary_crossentropy for binary classification
    • categorical_crossentropy for multi-class
  • Optimizer: Adam

πŸ“ˆ Evaluation

  • Accuracy
  • Confusion Matrix
  • Loss and Accuracy curves (training vs validation)