training & prediction work

need to work on assessment metrics & visualization