o qDe@s8ddlZddlZddlZeeZddZdddZdS)Nc CsFi}|D]\}}||g}||g}|||d}|||<q|S)N) courses_data users_datacompetency_data)itemsget) rrrtrained_models client_id client_dataclient_courses_dataclient_competency_data trained_modelr C/var/www/chatrigo.edurigo1.com/get_recommendations/model_trainer.py train_models   rrc Csz=|D]5\}}tj|d|d}tjtj|ddt|d}t||Wdn1s5wYqWdSt yP}zt d|d}~ww)Nmodel_z.pklT)exist_okwbzFailed to save model) rospathjoinmakedirsdirnameopenpickledump Exceptionlogger exception)r directoryrr model_pathfileer r r save_models  r")r)rrlogging getLogger__name__rrr"r r r rs