- improved RnnStack
- AStack: context doesn't belong o training data git-svn-id: http://moon:8086/svn/software/trunk/projects/Rbm@827 b431acfa-c32f-4a4a-93f1-934dc6c82436
This commit is contained in:
+14
-1
@@ -18,6 +18,8 @@
|
||||
|
||||
class RnnStack : public AStack
|
||||
{
|
||||
const size_t NUM_CODES = 37;
|
||||
|
||||
public:
|
||||
RnnStack(const std::string &name, size_t numContext);
|
||||
RnnStack(const RnnStack& orig) = delete;
|
||||
@@ -34,9 +36,20 @@ public:
|
||||
arma::mat vc_to_v(const arma::mat &vc) const;
|
||||
arma::mat vc_to_c(const arma::mat &vc) const;
|
||||
|
||||
arma::mat step_forward(arma::mat &state, const arma::mat &v);
|
||||
|
||||
arma::mat to_next(const arma::mat &v)
|
||||
{
|
||||
return v.cols(0, NUM_CODES-1);
|
||||
}
|
||||
|
||||
arma::mat to_curr(const arma::mat &v)
|
||||
{
|
||||
return v.cols(NUM_CODES, 2*NUM_CODES-1);
|
||||
}
|
||||
|
||||
private:
|
||||
size_t m_numContext;
|
||||
|
||||
};
|
||||
|
||||
#endif /* RNNSTACK_HPP */
|
||||
|
||||
Reference in New Issue
Block a user