To run the code one needs first download and unzip the
store.csv files on Kaggle and put them in this folder.
The following packages are needed 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 keras
Please refer to Keras for more details regarding how to install keras.
Next, run the following scripts to extract the csv files and prepare the 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: