Yes you’re right! If you try the code on another dataset please share the results, I’d be curious to see