improved prediction cmap plot, still searching for network parameters
small delays --> Lindgren