CategoricalTransformationParameter.cs

using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.ComponentModel;
using MyCaffe.basecode;
 
namespace MyCaffe.param.tft
{
    [Serializable]
    [TypeConverter(typeof(ExpandableObjectConverter))]
    public class CategoricalTransformationParameter : LayerParameterBase
    {
        uint m_nNumInput = 0;
        uint m_nStateSize = 0;
        List<int> m_rgCardinalities = new List<int>();
 
        public CategoricalTransformationParameter()
        {
        }
 
        [Description("The number of categorical inputs for the layer.")]
        public uint num_input
        {
            get { return m_nNumInput; }
            set { m_nNumInput = value; }
        }
 
        [Description("The state size that defines the output embedding dimension width.")]
        public uint state_size
        {
            get { return m_nStateSize; }
            set { m_nStateSize = value; }
        }
 
        [Description("The cardinalities specify the quantity of categories associated with each of the input variables.")]
        public List<int> cardinalities
        {
            get { return m_rgCardinalities; }
            set { m_rgCardinalities = value; }
        }
 
        public override object Load(System.IO.BinaryReader br, bool bNewInstance = true)
        {
            RawProto proto = RawProto.Parse(br.ReadString());
            CategoricalTransformationParameter p = FromProto(proto);
 
            if (!bNewInstance)
                Copy(p);
 
            return p;
        }
 
        public override void Copy(LayerParameterBase src)
        {
            CategoricalTransformationParameter p = (CategoricalTransformationParameter)src;
 
            m_nNumInput = p.m_nNumInput;
            m_nStateSize = p.m_nStateSize;
            m_rgCardinalities = new List<int>(p.m_rgCardinalities);
        }
 
        public override LayerParameterBase Clone()
        {
            CategoricalTransformationParameter p = new CategoricalTransformationParameter();
            p.Copy(this);
            return p;
        }
 
        public override RawProto ToProto(string strName)
        {
            RawProtoCollection rgChildren = new RawProtoCollection();
 
            rgChildren.Add("num_input", num_input.ToString());
            rgChildren.Add("state_size", state_size.ToString());
            rgChildren.Add<int>("cardinality", m_rgCardinalities);
 
            return new RawProto(strName, "", rgChildren);
        }
 
        public static CategoricalTransformationParameter FromProto(RawProto rp)
        {
            string strVal;
            CategoricalTransformationParameter p = new CategoricalTransformationParameter();
 
            if ((strVal = rp.FindValue("num_input")) != null)
                p.num_input = uint.Parse(strVal);
 
            if ((strVal = rp.FindValue("state_size")) != null)
                p.state_size = uint.Parse(strVal);
 
            p.cardinalities = rp.FindArray<int>("cardinality");
 
            return p;
        }
    }
}