/News-Data-Classification

Using the Reuters dataset, this example illustrates the process of data preprocessing, model definition and training, and performance evaluation.

Primary LanguageJupyter NotebookMIT LicenseMIT

News-Data-Classification

Overview

Using the Reuters dataset, this example illustrates the process of data preprocessing, model definition and training, and performance evaluation.

Project Structure

  • News_Data_Classification.ipynb: Jupyter notebook file containing the code for building and training the neural-network model using Reuters dataset for classification.

Results

The model achieved an accuracy of ~ 82% on the test set. This is a good accuracy for this use case and the model can be further improved by tuning the hyperparameters.

Installation

To get started, clone the repository and install the required dependencies

In Docker and JupyterLab

git clone https://github.com/lucianoscarpaci/News-Data-Classification

License

This project is licensed under the MIT License.