fix to pull request #54...

scalanlp · Apr 19, 2013 · 4f79aef · 4f79aef
1 parent 4b0f1f7
commit 4f79aef
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/learn/src/main/scala/breeze/classify/NNetClassifier.scala b/learn/src/main/scala/breeze/classify/NNetClassifier.scala
@@ -24,10 +24,10 @@ class NNetClassifier[L, T](nnet: NeuralNetwork,
 }
 
 object NNetClassifier {
-  class CounterTrainer[L, T](opt: OptParams = OptParams()) extends Classifier.Trainer[L, Counter[T, Double]] {
+  class CounterTrainer[L, T](opt: OptParams = OptParams(),layersIn:Array[Int] = Array(100)) extends Classifier.Trainer[L, Counter[T, Double]] {
     type MyClassifier = NNetClassifier[L, Counter[T, Double]]
 
-    def train(data: Iterable[Example[L, Counter[T, Double]]],layersIn:Array[Int] = Array(100)) = {
+    def train(data: Iterable[Example[L, Counter[T, Double]]]) = {
       val labels = Index[L]()
       data foreach { labels index _.label}
       val featureIndex = Index[T]()
@@ -48,7 +48,7 @@ object NNetClassifier {
       val obj = new NNObjective(processedData.toIndexedSeq, errorFun, layers)
       val guess = obj.initialWeightVector
       val weights = opt.minimize(obj,guess)
-      new NNetClassifier(obj.extract(weights), {fEncoder.encodeDense(_, true)}, labels)
+      new NNetClassifier(obj.extract(weights), {fEncoder.encodeDense(_:Counter[T, Double], true)}, labels)
     }
   }
 }
diff --git a/project/build.properties b/project/build.properties
@@ -1 +1 @@
-sbt.version=0.12.0
+sbt.version=0.12.3