Neural Network Model
Collection
1 item • Updated
YAML Metadata Warning:empty or missing yaml metadata in repo card
Check out the documentation for more information.
Simple ANN Binary Classification Model
Overview
This project implements an Artificial Neural Network (ANN) using TensorFlow/Keras for binary classification.
Model Architecture
- Input features: 2
- Hidden layers:
- Dense layer (6 neurons, ReLU)
- Dense layer (6 neurons, ReLU)
- Output:
- Dense layer (1 neuron, Sigmoid)
Training
- Optimizer: Adam
- Loss: Binary Crossentropy
- Epochs: 100
- Batch size: 32
Files
- simpleArtificialNeuralNetworkTraining.py: Model training
- evaluation.py: Model evaluation
- inference.py: Prediction script
- model.keras: Trained model
- Downloads last month
- 85
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support