docs(examples): use built-in train-test split code

[jitingxu1]

replace customized train test split code to the train_test_split function

[codecov-commenter]

Codecov Report

All modified and coverable lines are covered by tests ✅

Project coverage is 88.70%. Comparing base (6dce35e) to head (37b66bf).

Additional details and impacted files

@@           Coverage Diff           @@
##             main     #172   +/-   ##
=======================================
  Coverage   88.70%   88.70%           
=======================================
  Files          27       27           
  Lines        2009     2009           
=======================================
  Hits         1782     1782           
  Misses        227      227           

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

[deepyaman]

Interestingly, your reproducible train-test split changed. 😂 Not looking into it; I assume it's because the logic is written differently in the function.

[jitingxu1]

changed

Same logic but with different random_seed