BostonModel

gistfile1.py

import pandas as pd
from sklearn.ensemble import RandomForestRegressor
import pickle
import datetime

## load data

df = pd.read_csv('/content/odsc-ml-drum/data/boston_housing.csv')
df.head()

## set features and target

X = df.drop('MEDV', axis=1)
y = df['MEDV']

## train the model
rf = RandomForestRegressor(n_estimators = 20)
rf.fit(X,y)

## serialize the model

with open('/content/odsc-ml-drum/src/custom_model/rf.pkl', 'wb') as pkl:
    pickle.dump(rf, pkl)