Added optional 3rd arg for dataset

src/main.cpp

@@ -27,8 +27,9 @@ vector <string> parseFeatures(string fileName) {
 
 int main(int argc, char *argv[]) {
 
-    if (argc <= 1) {
-        cout << "arguments needed\n";
+    if ((argc <= 2) || (argc >= 5)) {
+        cout << "Given " << to_string(argc) <<" args. Need 2 or 3." << endl;
+		cout << "race [numTrees] [depth] [dataset (no suffix, optional)]" << endl;
         exit(1);
     }
 
@@ -43,9 +44,18 @@ int main(int argc, char *argv[]) {
     vector <vector<string>> datasetAsString;
     vector <FeatureType> featureTypes;
     vector <string> features;
+	if (argc == 4) {
+		cout << "Dataset: " << argv[3] << endl;
+		datasetAsString = parseDataToString( (string)argv[3] + ".data");
+   		featureTypes = parseFeatureTypes( (string)argv[3] + ".featureTypes");
+	    features = parseFeatures( (string)argv[3] + ".features");
+	}
+	else {
+		cout << "WARNING: No dataset provided as an argument!" << endl;
 	    datasetAsString = parseDataToString("../datasets/adult.data");
    		featureTypes = parseFeatureTypes("../datasets/adult.featureTypes");
 	    features = parseFeatures("../datasets/adult.features");
+	}
 
     //pick number of features to select for random sub-spacing
     float featureWeight = sqrt(features.size())/features.size();