Rewrite model if fit is being called after partial_fit or vice versa