RFrunwith_

bin/functions/hibench_prop_env_mapping.py

@@ -119,12 +119,12 @@
     # For Random Forest
     NUM_EXAMPLES_RF="hibench.rf.examples",
     NUM_FEATURES_RF="hibench.rf.features",
-    NUMTREES="hibench.rf.numTrees",
-    NUMCLASSES="hibench.rf.numClasses",
-    FEATURESUBSETSTRATEGY="hibench.rf.featureSubsetStrategy",
-    IMPURITY="hibench.rf.impurity",
-    MAXDEPTH="hibench.rf.maxDepth",
-    MAXBINS="hibench.rf.maxBins",
+    NUM_TREES_RF="hibench.rf.numTrees",
+    NUM_CLASSES_RF="hibench.rf.numClasses",
+    FEATURE_SUBSET_STRATEGY_RF="hibench.rf.featureSubsetStrategy",
+    IMPURITY_RF="hibench.rf.impurity",
+    MAX_DEPTH_RF="hibench.rf.maxDepth",
+    MAX_BINS_RF="hibench.rf.maxBins",
     # For SVD
     NUM_EXAMPLES_SVD="hibench.svd.examples",
     NUM_FEATURES_SVD="hibench.svd.features",

bin/workloads/ml/rf/spark/run.sh

@@ -26,12 +26,12 @@ rmr_hdfs $OUTPUT_HDFS || true
 
 SIZE=`dir_size $INPUT_HDFS`
 START_TIME=`timestamp`
-OPTION="--numTrees $NUMTREES \
-        --numClasses $NUMCLASSES \
-        --featureSubsetStrategy $FEATURESUBSETSTRATEGY \
-        --impurity $IMPURITY \
-        --maxDepth $MAXDEPTH \
-        --maxBins $MAXBINS"
+OPTION="--numTrees $NUM_TREES_RF \
+        --numClasses $NUM_CLASSES_RF \
+        --featureSubsetStrategy $FEATURE_SUBSET_STRATEGY_RF \
+        --impurity $IMPURITY_RF \
+        --maxDepth $MAX_DEPTH_RF \
+        --maxBins $MAX_BINS_RF"
 run_spark_job com.intel.hibench.sparkbench.ml.RandomForestClassification $OPTION $INPUT_HDFS
 END_TIME=`timestamp`