Dataset shift | Python