fixing FitClusterer to work with fit_predict