updates

samindaa · Feb 2, 2016 · 37e2f98 · 37e2f98
1 parent 7ef7c87
commit 37e2f98
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 4 deletions.
diff --git a/include/Function.h b/include/Function.h
@@ -47,7 +47,6 @@ namespace RLLib
       }
       virtual T initialize() =0;
       virtual void reset() =0;
-      virtual Vector<T>* weights() const =0;
   };
 
   template<typename T>

diff --git a/include/Predictor.h b/include/Predictor.h
@@ -36,6 +36,7 @@ namespace RLLib
       {
       }
       virtual T predict(const Vector<T>* x) const =0;
+      virtual Vector<T>* weights() const =0;
   };
 
   template<typename T>

diff --git a/test/GQTest.cpp b/test/GQTest.cpp
@@ -104,9 +104,7 @@ void GQTest::testGQOnRandomWalk(const double& targetLeftProbability,
   Assert::assertPasses(nbEpisode > 100);
 
   const Predictor<double>* predictor = learner->predictor();
-  const LinearLearner<double>* gqLearner =
-      static_cast<const LinearLearner<double>*>(reinterpret_cast<const GQ<double>*>(predictor));
-  const Vector<double>* v = gqLearner->weights();
+  const Vector<double>* v = predictor->weights();
   Assert::checkValues(v);
 
   double error = FiniteStateGraph::distanceToSolution(&solution, vFun);