Add some doc for the new partial_fit method