Make crossvalidate compatible with new networks

New networks use 2 outputs to match survival data

Signed-off-by: Jonas Kalderstam <jonas@kalderstam.se>