Eligma Product Category Predictor
Output used as a new feature in Discovery Engine and as a guide for extracting other product features.
Deep convolutional neural network for category classification.
This is early prototype, proof-of-concept and under development.
Preprocessed data of products and categories.
nltk==3.2.5 pandas==0.22.0 tensorflow==1.4.1 numpy==1.14.0 h5py==2.7.1 scikit_learn==0.19.1
pip3 install --user -r requirements.txt
Performance on test set
accuracy = 0.96865145 auroc = 0.9882318
Currently we do not accept pull requests and contributions.
- Bostjan Mrak
This project is licensed under the MIT License.
- Eligma Team