fix(models,traininig): hierarchical model + integration test (#400)

fix(models,traininig): hierarchical model + integration test (#400) #1