Guest User

Untitled

a guest
May 24th, 2018
82
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 1.24 KB | None | 0 0
  1. # reading training data
  2. zf_test = zipfile.ZipFile('C:/Users/Umar/Downloads/test.csv.zip')
  3. df_test = pd.read_csv(zf_test.open('test.csv'), converters={'POLYLINE': lambda x: json.loads(x)[-1:]})
  4. latlong_test = np.array([[p[0][1], p[0][0]] for p in df_test['POLYLINE'] if len(p)>0])
  5.  
  6. zf_train = zipfile.ZipFile('C:/Users/Umar/Downloads/train.csv.zip')
  7. df_train = pd.read_csv(zf_train.open('train.csv'), converters={'POLYLINE': lambda x: json.loads(x)[-1:]})
  8. latlong_train = np.array([[p[0][1], p[0][0]] for p in df_train['POLYLINE'] if len(p)>0])
  9. # cut off long distance trips
  10. lat_low, lat_hgh = np.percentile(latlong_train[:,0], [2, 98])
  11. lon_low, lon_hgh = np.percentile(latlong_train[:,1], [2, 98])
  12.  
  13.  
  14.  
  15. # create image
  16. bins = 513
  17. lat_bins = np.linspace(lat_low, lat_hgh, bins)
  18. lon_bins = np.linspace(lon_low, lon_hgh, bins)
  19. H2, _, _ = np.histogram2d(latlong_train[:,0], latlong_train[:,1], bins=(lat_bins, lon_bins))
  20.  
  21. prior = H2 + np.ones(shape=prior.shape)
  22. prior = prior/np.sum(prior,dtype=np.float64)
  23.  
  24. lat_centroids = [(lat_bins[i]+lat_bins[i+1])/2.0 for i in xrange(len(lat_bins)-1)]
  25. lon_centroids = [(lon_bins[i]+lon_bins[i+1])/2.0 for i in xrange(len(lon_bins)-1)]
  26.  
  27. name 'prior' is not defined
Add Comment
Please, Sign In to add comment