HelloI want answer for my question. I have python code that I need to add train$\_$test$\_$split function plz help me $($using my code in down$)$

import numpy as npfrom sklearn.datasets import load$\_$diabetes diabetes$\_$X$,$ diabetes$\_$y $=$ load$\_$diabetes $($return$\_$X$\_$y$=$ True$)$

diabetes$\_$X$[0$:$5]$

diabetes$\_$y$[0$:$5]$

diabetes$\_$X$.$shape

diabetes$\_$X $=$ diabetes$\_$X$[$:$,$np$.$newaxis,$2]$

diabetes$\_$X$.$shape diabetes$\_$X$\_$train $=$ diabetes$\_$X$[$:$-20]$

diabetes$\_$X$\_$test $=$ diabetes$\_$X$[-20$:$]$ diabetes$\_$X$\_$train.shape

diabetes$\_$X$\_$test.shape

diabetes$\_$y$\_$train $=$ diabetes$\_$y$[$:$-20]$diabetes$\_$y$\_$test $=$ diabetes$\_$y$[-20$:$]$ diabetes$\_$y$\_$train.shape

diabetes$\_$y$\_$test.shape