itaynir1 / tf-second_car_hand

The code builds a machine learning model to predict the price of second-hand cars. It preprocesses the data, defines a neural network model, and trains it using the training dataset. The model's performance is evaluated, and predictions are made on new car data.

Home Page:https://github.com/itaynir1/tf-second_car_hand

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

SECOND HAND CARS DATA SET

Regression problem - using TensorFlow

Description:

This code is a script for building and training a machine-learning model for predicting the price of second-hand cars.

This code uses libraries such as TensorFlow, pandas, seaborn, and matplotlib for data processing, visualization, and model building.

Data preparation is performed by reading a CSV file ("train.csv") containing car features and labels, and then exploring the data using visualizations.

The data is normalized using the Normalization layer from TensorFlow.
The model architecture is defined using the Sequential API from Keras, consisting of several dense layers with relu activation and a final output layer.
The model is compiled with an optimizer (Adam), loss function (MeanAbsoluteError), and metrics (RootMeanSquaredError).

The model is trained using the training dataset and validated using the validation dataset. Training history and performance metrics are plotted.

Finally, the trained model is evaluated on the test dataset, and predictions are made on new car data.

About

The code builds a machine learning model to predict the price of second-hand cars. It preprocesses the data, defines a neural network model, and trains it using the training dataset. The model's performance is evaluated, and predictions are made on new car data.

https://github.com/itaynir1/tf-second_car_hand


Languages

Language:Jupyter Notebook 99.8%Language:Python 0.2%