class_docstr: y = a * x + b
class_docstr_tok: ["y","=","a","*","x","+","b"]
code: def __getitem__(self, index):
      return self.x[index, None], self.y[index, None]
  
imports: import os
  from unittest import TestCase
  import numpy
  import pytest
  import torch
  import torch.nn
  from zoo.orca import OrcaContext
  from zoo.orca.data.pandas import read_csv
  from zoo.orca.learn.metrics import Accuracy
  from zoo import init_nncontext
  from zoo.orca.learn.pytorch import Estimator
  from zoo.orca.data import SparkXShards
  from zoo.orca.data.image.utils import chunks
label_desc: Use image node_id 2 for calling the LinearDataset obj's underlying member method code with example usage: obj.__getitem__(index) and returns: self,  self