BotSharp/BotSharp.NLP/Models/CRFLite/CRFDecoder.cs

/*
 * BotSharp.NLP Library
 * Copyright (C) 2018 Haiping Chen
 * 
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 * 
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 * 
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

using BotSharp.Models.CRFLite.Decoder;
using System;
using System.Collections.Generic;
using System.IO;
using System.Text;

namespace BotSharp.Models.CRFLite
{
    public class CRFDecoder
    {
        ModelReader _modelReader;

        /// <summary>
        /// Load encoded model from file
        /// </summary>
        /// <param name="modelFilename">
        /// The model path.
        /// </param>
        /// <returns></returns>
        public void LoadModel(string modelFilename)
        {
            _modelReader = new ModelReader(modelFilename);
            _modelReader.LoadModel();
        }

        /// <summary>
        /// Loads an encoded model using the specified delegate.
        /// Using this overload you can read the model e.g. 
        /// from network, zipped archives or other locations, as you wish.
        /// </summary>
        /// <param name="modelLoader">
        /// Allows reading the model from arbitrary formats and sources.
        /// </param>
        /// <param name="modelFilename">
        /// The model file name, as used by the given <paramref name="modelLoader"/>
        /// for file resolution.
        /// </param>
        /// <returns></returns>
        public void LoadModel(Func<string, Stream> modelLoader, string modelFilename)
        {
            this._modelReader = new ModelReader(modelLoader, modelFilename);
            _modelReader.LoadModel();
        }

        public SegDecoderTagger CreateTagger(int nbest, int this_crf_max_word_num = BaseUtils.DEFAULT_CRF_MAX_WORD_NUM)
        {
            if (_modelReader == null)
            {
                return null;
            }

            var tagger = new SegDecoderTagger(nbest, this_crf_max_word_num);
            tagger.init_by_model(_modelReader);

            return tagger;
        }

        //Segment given text
        public int Segment(crf_seg_out[] pout, //segment result
            SegDecoderTagger tagger, //Tagger per thread
            List<List<string>> inbuf //feature set for segment
            )
        {
            var ret = 0;
            if (inbuf.Count == 0)
            {
                //Empty input string
                return BaseUtils.RETURN_SUCCESS;
            }

            ret = tagger.reset();
            if (ret < 0)
            {
                return ret;
            }

            ret = tagger.add(inbuf);
            if (ret < 0)
            {
                return ret;
            }

            //parse
            ret = tagger.parse();
            if (ret < 0)
            {
                return ret;
            }

            //wrap result
            ret = tagger.output(pout);
            if (ret < 0)
            {
                return ret;
            }

            return BaseUtils.RETURN_SUCCESS;
        }



        //Segment given text
        public int Segment(CRFTermOut[] pout, //segment result
            DecoderTagger tagger, //Tagger per thread
            List<List<string>> inbuf //feature set for segment
            )
        {
            var ret = 0;
            if (inbuf.Count == 0)
            {
                //Empty input string
                return BaseUtils.RETURN_SUCCESS;
            }

            ret = tagger.reset();
            if (ret < 0)
            {
                return ret;
            }

            ret = tagger.add(inbuf);
            if (ret < 0)
            {
                return ret;
            }

            //parse
            ret = tagger.parse();
            if (ret < 0)
            {
                return ret;
            }

            //wrap result
            ret = tagger.output(pout);
            if (ret < 0)
            {
                return ret;
            }

            return BaseUtils.RETURN_SUCCESS;
        }
    }
}