code: @Override
      public void fit(org.nd4j.linalg.dataset.api.DataSet data) {
          fit(data.getFeatures(), data.getLabels(), data.getFeaturesMaskArray(), data.getLabelsMaskArray());
      }
docs: Fit the model for one iteration on the provided data
  
  @param data the data to train on