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);
}