Eval label exception fix

src/main/scala/junto/Junto.scala

@@ -29,7 +29,7 @@ object Junto {
     conf.evalLabelFile.toOption match {
 
       case Some(evalLabelFile) => {
-        val evalLabelSequence = getLabels(evalLabelFile, separator)
+        val evalLabelSequence = getLabels(evalLabelFile, skipHeader = true)
 
         val evalLabels = (for {
           LabelSpec(nodeName, label, strength) <- evalLabelSequence
@@ -45,6 +45,7 @@ object Junto {
     }
 
     // Output predictions if an output file is specified.
+    // output predictions are comma seperated
     conf.outputFile.toOption match {
       case Some(outputFile) =>
 

src/main/scala/junto/graph/package.scala

@@ -39,15 +39,19 @@ package object graph {
     }
   }
 
-  def getLabels(inputFile: String, separator: Char = ','): Iterator[LabelSpec] = {
+  def getLabels(inputFile: String, separator: Char = ',', skipHeader: Boolean = false): Iterator[LabelSpec] = {
+    val lines = getSource(inputFile).getLines
+
+    val header = if (skipHeader) lines.next() else None
+    // skip the header in eval files for now (perhaps use it later to identify labels with probabilities)
     for {
-      line <- getSource(inputFile).getLines
+      line <- lines
       items = line.split(separator)
     } yield {
-      val source = items(0)
-      val target = items(1)
+      val node = items(0)
+      val label = items(1)
       val weight = if (items.length == 2) 1.0 else items(2).toDouble
-      LabelSpec(source, target, weight)
+      LabelSpec(node, label, weight)
     }
   }