BoundsError: attempt to access n-element UnitRange{Int32} at index [n+1]

found the error, forgot to change the outputlayer to the size of number of datasets.