code: def __init__(self, size=1000, a=2, b=5):
      x, y = get_linear_data(a, b, size)
      self.x = torch.from_numpy(x.reshape(-1, 1))
      self.y = torch.from_numpy(y.reshape(-1, 1))
  
imports: from unittest import TestCase
  from zoo.orca.automl.model.base_pytorch_model import PytorchModelBuilder
  from torch.utils.data import Dataset, DataLoader
  import pytest
  import numpy
  import torch
  import torch.nn
label_desc: Use image node_id 1 to create a new LinearDataset object from inherited base classes: Dataset with example: obj = LinearDataset(size, a, b)