Skip to content

ismaile17/machine-learning

Folders and files

NameName
Last commit message
Last commit date

Latest commit

Β 

History

2 Commits
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 

Repository files navigation

πŸ“Š Machine Learning Tabular Model (Keras + TensorFlow)

A simple but effective machine learning pipeline built with TensorFlow/Keras for training on structured/tabular data using a CSV file as input.


πŸš€ Features

  • βœ… Automatic encoding and normalization for numeric and categorical features
  • 🧠 Dense Neural Network (DNN) architecture
  • πŸ“¦ TensorFlow tf.data.Dataset pipeline
  • πŸ”Ž Model visualization with keras.utils.plot_model
  • πŸ“‘ CSV-based training data
  • πŸ§ͺ Separate training (main.py) and testing (test.py) scripts

πŸ“ Project Structure

File/Folder Description
main.py Trains the model and processes the dataset
test.py Loads the model and makes predictions
data.csv Tabular dataset including input features + target
Data/ (Optional) Additional files or output directory

πŸ” Technical Notes Numeric features are normalized using Normalization layers.

Categorical features are encoded using StringLookup.

The target column must be named Target in the CSV.

Dataset is processed using the efficient tf.data.Dataset pipeline.

The model structure can be visualized using plot_model.

πŸ”§ How to Run

1. Install dependencies:

pip install tensorflow pandas numpy
python main.py
python test.py


About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages