Imported from https://github.com/AlexMoreo/learning-to-weight