Loading data for the regression example

data_load_reg_pytorch.py

data_name ='medical_insurance'

# Read in the medical insurance data
df = pd.read_csv('https://raw.githubusercontent.com/StatsGary/Data/main/insurance.csv')

# Drop nulls
df.dropna(axis='columns',inplace=True)

# Get number of rows
obs = len(df)

# Divide obs in half to get half batch size
batch_size = obs // 2