To run the code one needs first download and unzip the
store.csv files on Kaggle and put them inside this folder.
The following packages is need if you want to recover the result in the paper (we used python 3):
pip3 install -U scikit-learn pip3 install -U xgboost pip3 install -U keras
Please refer to Keras for more details regarding isntalling keras.
Next run the following scripts to extract csv files and prepare features:
python3 extract_csv_files.py python3 prepare_features.py
To run the models:
You can anaylize the embeddings with the ipython notebook included. This is the learned embeeding of German States printed in 2D (with the Kaggle branch):
and this is the learned embeddings of 1115 Rossmann stores printed in 3D: