Back to Projects
2024
Project Detail
Category Classification
A machine learning category classification model based on product names using LSTM.
Worked at
Bangkit Academy (Capstone/ML Track)
Role / Main Duties
Machine Learning Cohort
Dilihat
0views
Tech Stack
PythonTensorFlowLSTMFlask
Project Description
This project focuses on building a text-based LSTM model for product category classification. The goal is to help automate product grouping so that data entry and analysis become faster.
Highlights
- Text preprocessing and tokenization
- LSTM model training and evaluation
- Parameter experiments to improve accuracy
Main Responsibilities
- Prepared the dataset and preprocessing (cleaning, tokenizing, padding).
- Built and trained the LSTM model in TensorFlow.
- Evaluated metrics and iterated the model for better performance.
- Built an API to integrate the model into the capstone app (in collaboration with the Mobile/Cloud teams).