FitTransformer class