Add predict
Merge branch 'groups_on_layer' of github.com:allenbo/distnet into groups_on_layer