code: @Override
      public void fit(@NonNull MultiDataSet dataSet) {
          Map<Integer, NormalizerStats.Builder> inputStatsBuilders = new HashMap<>();
          Map<Integer, NormalizerStats.Builder> outputStatsBuilders = new HashMap<>();
  
          fitPartial(dataSet, inputStatsBuilders, outputStatsBuilders);
  
          inputStats = buildAllStats(inputStatsBuilders);
          outputStats = buildAllStats(outputStatsBuilders);
      }
docs: Fit a MultiDataSet (only compute based on the statistics from this dataset)
  
  @param dataSet the dataset to compute on