code: def __init__(self, size=1000):
      x, y = get_x_y(size=size)
      self.x = torch.from_numpy(x).float()
      self.y = torch.from_numpy(y).float()
  
imports: import os
  from unittest import TestCase
  import numpy
  import pytest
  import torch
  import torch.nn
  from torch.utils.data import Dataset, DataLoader
  from zoo.orca.automl.auto_estimator import AutoEstimator
  from zoo.orca.automl.pytorch_utils import LR_NAME
  from zoo.orca.automl import hp
label_desc: Use image node_id 1 to create a new CustomDataset object from inherited base classes: Dataset with example: obj = CustomDataset(size)