added script to just learn the embedding