handle nan values

matsim-org · May 10, 2024 · 50b4ce9 · 50b4ce9
1 parent 51ebdfa
commit 50b4ce9
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/contribs/application/src/main/java/org/matsim/application/prepare/Predictor.java b/contribs/application/src/main/java/org/matsim/application/prepare/Predictor.java
@@ -11,6 +11,7 @@ public interface Predictor {
 
 	/**
 	 * Predict value from given features.
+	 * @return predicted value, maybe NaN if no prediction is possible.
 	 */
 	double predict(Object2DoubleMap<String> features, Object2ObjectMap<String, String> categories);
 

diff --git a/...ation/src/main/java/org/matsim/application/prepare/network/params/ApplyNetworkParams.java b/...ation/src/main/java/org/matsim/application/prepare/network/params/ApplyNetworkParams.java
@@ -147,6 +147,9 @@ private void applyChanges(Link link, Feature ft) {
 			}
 
 			double perLane = capacity.predict(ft.features(), ft.categories());
+			if (Double.isNaN(perLane)) {
+				return;
+			}
 
 			double cap = capacityEstimate(ft.features().getDouble("speed"));
 
@@ -181,6 +184,10 @@ private void applyChanges(Link link, Feature ft) {
 				speedModel.predict(ft.features(), ft.categories(), paramsOpt.getParams(ft.junctionType())) :
 				speedModel.predict(ft.features(), ft.categories());
 
+			if (Double.isNaN(speedFactor)) {
+				return;
+			}
+
 			if (speedFactor > speedFactorBounds[1]) {
 				log.warn("Reducing speed factor on {} from {} to {}", link.getId(), speedFactor, speedFactorBounds[1]);
 				speedFactor = speedFactorBounds[1];