Switch branches/tags
Nothing to show
Find file History
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Type Name Latest commit message Commit time
..
Failed to load latest commit information.
README.md
dataFeed.py
model.py
requirements.txt
train.py
utils.py

README.md

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.

Prerequisites

Preprocessed data of products and categories.

Python 3.5.

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

Install

pip3 install --user -r requirements.txt

Performance on test set

accuracy = 0.96865145 auroc = 0.9882318

Contributing

Currently we do not accept pull requests and contributions.

Authors

  • Bostjan Mrak

License

This project is licensed under the MIT License.

Acknowledgments

  • Eligma Team
  • TensorFlow