public RNNDecoder(string strModelFileName, Featurizer featurizer) { MODELDIRECTION modelDir = MODELDIRECTION.FORWARD; RNNHelper.CheckModelFileType(strModelFileName, out modelDir); if (modelDir == MODELDIRECTION.BI_DIRECTIONAL) { Logger.WriteLine("Model Structure: Bi-directional RNN"); m_Rnn = new BiRNN(); } else { Logger.WriteLine("Model Structure: Simple RNN"); m_Rnn = new ForwardRNN(); } m_Rnn.LoadModel(strModelFileName); Logger.WriteLine("CRF Model: {0}", m_Rnn.IsCRFTraining); m_Featurizer = featurizer; }
public RNNDecoder(string strModelFileName) { MODELDIRECTION modelDir = MODELDIRECTION.FORWARD; MODELTYPE modelType; RNNHelper.CheckModelFileType(strModelFileName, out modelDir, out modelType); if (modelDir == MODELDIRECTION.BI_DIRECTIONAL) { Logger.WriteLine("Model Structure: Bi-directional RNN"); rnn = new BiRNN <Sequence>(); } else { Logger.WriteLine("Model Structure: Simple RNN"); rnn = new ForwardRNN <Sequence>(); } ModelType = modelType; rnn.LoadModel(strModelFileName); Logger.WriteLine("CRF Model: {0}", rnn.IsCRFTraining); }