A simple but effective machine learning pipeline built with TensorFlow/Keras for training on structured/tabular data using a CSV file as input.
- β Automatic encoding and normalization for numeric and categorical features
- π§ Dense Neural Network (DNN) architecture
- π¦ TensorFlow
tf.data.Datasetpipeline - π Model visualization with
keras.utils.plot_model - π CSV-based training data
- π§ͺ Separate training (
main.py) and testing (test.py) scripts
| 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.
pip install tensorflow pandas numpy
python main.py
python test.py