Refactoring is_fit_stage logic
Created by: valer1435
Due to single predict method for fi and predict stage methods in implementation become complicated. I propose to change current behaviour by dividing method predict on two methods predict
and predict_for_fit
.