public RNNDecoder(string strModelFileName, Featurizer featurizer)
{
MODELTYPE modelType = MODELTYPE.SIMPLE;
MODELDIRECTION modelDir = MODELDIRECTION.FORWARD;
RNN.CheckModelFileType(strModelFileName, out modelType, out modelDir);
if (modelDir == MODELDIRECTION.BI_DIRECTIONAL)
{
Logger.WriteLine("Model Structure: Bi-directional RNN");
if (modelType == MODELTYPE.SIMPLE)
{
m_Rnn = new BiRNN(new SimpleRNN(), new SimpleRNN());
}
else
{
m_Rnn = new BiRNN(new LSTMRNN(), new LSTMRNN());
}
}
else
{
if (modelType == MODELTYPE.SIMPLE)
{
Logger.WriteLine("Model Structure: Simple RNN");
m_Rnn = new SimpleRNN();
}
else
{
Logger.WriteLine("Model Structure: LSTM-RNN");
m_Rnn = new LSTMRNN();
}
}
m_Rnn.LoadModel(strModelFileName);
Logger.WriteLine("CRF Model: {0}", m_Rnn.IsCRFTraining);
m_Featurizer = featurizer;
}
