code: def split_train_test(data):
      RANDOM_STATE = 2021
      train, test = train_test_split(
          data, test_size=0.2, random_state=RANDOM_STATE
      )
      return train, test
  
file_path: BigDL/python/orca/tutorial/xshards/tabular_playground_series.py
imports: from sklearn.model_selection import train_test_split
  import torch
  import torch.nn
  import torch.optim
  from bigdl.orca import init_orca_context, stop_orca_context
  import bigdl.orca.data.pandas
  from bigdl.orca.data.transformer import *
  from bigdl.orca.learn.pytorch import Estimator
  from bigdl.orca.learn.metrics import Accuracy
label_desc: Use image node_id 3 for calling a global function with example usage: split_train_test(data) and returns: train,  test