use sklearn to solve