embed rossman data

embedder_demo.py

import pandas as pd
from embedder.regression import Embedder
from embedder.preprocessing import (categorize, 
     pick_emb_dim, encode_categorical)

rossman = pd.read_csv('rossman.csv')
y = rossman['Sales']
X = rossman.drop('Sales', axis=1)

cat_vars = categorize(rossman)
embedding_dict = pick_emb_dim(cat_vars, max_dim=50)
X_encoded, encoders = encode_categorical(X)
embedder = Embedder(embedding_dict, model_json=None)
embedder.fit(X_encoded, y)