--- a/scripts/classify-objects.py	Wed Jul 12 13:14:09 2023 -0400
+++ b/scripts/classify-objects.py	Wed Jul 12 13:21:08 2023 -0400
@@ -37,9 +37,9 @@
 pedNorm = norm(classifierParams.meanPedestrianSpeed, classifierParams.stdPedestrianSpeed)
 # numpy lognorm shape, loc, scale: shape for numpy is scale (std of the normal) and scale for numpy is exp(location) (loc=mean of the normal)
 bicLogNorm = lognorm(classifierParams.scaleCyclistSpeed, loc = 0., scale = np.exp(classifierParams.locationCyclistSpeed))
-speedProbabilities = {'car': lambda s: carNorm.pdf(s),
-                      'pedestrian': lambda s: pedNorm.pdf(s), 
-                      'bicycle': lambda s: bicLogNorm.pdf(s)}
+speedProbabilities = {moving.userTypeNames[1]: lambda s: carNorm.pdf(s),
+                      moving.userTypeNames[2]: lambda s: pedNorm.pdf(s), 
+                      moving.userTypeNames[4]: lambda s: bicLogNorm.pdf(s)}
 
 if args.plotSpeedDistribution:
     import matplotlib.pyplot as plt