# calculate the target
df['insurance_benefits_received'] = (df['insurance_benefits']>0).astype(int)
x = df.drop(['insurance_benefits_received'], axis=1)
y = df['insurance_benefits_received']
x_train,x_test,y_train,y_test = train_test_split(x,y,test_size = 0.3,random_state=12)
Editor is loading...
