sivakornchong's picture
add code to calculate df
7c9d88c
raw
history blame contribute delete
No virus
538 Bytes
import json
import requests
from misc import nearest_mrt
import time
import pickle
import os
import pandas as pd
filename = 'finalized_model.sav'
if os.path.exists("./finalized_model.sav"):
model = pickle.load(open(filename, 'rb'))
print('loaded model')
else:
print('failed loading model')
#extract feature names#
feature_names = model.feature_names
site_names = feature_names[5:31]
town = []
for town1 in site_names:
town1 = town1[5:]
town.append(town1)
print(town)
print(len(town))
print(len(feature_names))