OneHotParameter.cs

using MyCaffe.basecode;
using System;
using System.Collections.Generic;
using System.ComponentModel;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
 
namespace MyCaffe.param.nt
{
    [Serializable]
    [TypeConverter(typeof(ExpandableObjectConverter))]
    public class OneHotParameter : LayerParameterBase
    {
        int m_nAxis = 2;
        uint m_nNumOutput = 16;
        double m_dfMin = -1.0;
        double m_dfMax = 1.0;
        int m_nMinAxes = 4;
 
        public OneHotParameter()
        {
        }
 
        [Description("Specifies the axis over which to apply the one-hot vectoring.")]
        public int axis
        {
            get { return m_nAxis; }
            set { m_nAxis = value; }
        }
 
        [Description("Specifies the minimum number of axes.  Axes of size 1 are added to the current axis count up to the minimum.")]
        public int min_axes
        {
            get { return m_nMinAxes; }
            set { m_nMinAxes = value; }
        }
 
        [Description("Specifies the number of items within the one-hot vector output.")]
        public uint num_output
        {
            get { return m_nNumOutput; }
            set { m_nNumOutput = value; }
        }
 
        [Description("Specifies the minimum data range over which to bucketize for the one-hot vector input.")]
        public double min
        {
            get { return m_dfMin; }
            set { m_dfMin = value; }
        }
 
        [Description("Specifies the maximum data range over which to bucketize for the one-hot vector input.")]
        public double max
        {
            get { return m_dfMax; }
            set { m_dfMax = value; }
        }
 
        public override object Load(System.IO.BinaryReader br, bool bNewInstance = true)
        {
            RawProto proto = RawProto.Parse(br.ReadString());
            OneHotParameter p = FromProto(proto);
 
            if (!bNewInstance)
                Copy(p);
 
            return p;
        }
 
        public override void Copy(LayerParameterBase src)
        {
            OneHotParameter p = (OneHotParameter)src;
 
            m_nAxis = p.m_nAxis;
            m_nNumOutput = p.m_nNumOutput;
            m_dfMin = p.m_dfMin;
            m_dfMax = p.m_dfMax;
            m_nMinAxes = p.m_nMinAxes;
        }
 
        public override LayerParameterBase Clone()
        {
            OneHotParameter p = new OneHotParameter();
            p.Copy(this);
            return p;
        }
 
        public override RawProto ToProto(string strName)
        {
            RawProtoCollection rgChildren = new RawProtoCollection();
 
            rgChildren.Add("axis", axis.ToString());
            rgChildren.Add("num_output", num_output.ToString());
            rgChildren.Add("min", min.ToString());
            rgChildren.Add("max", max.ToString());
            rgChildren.Add("min_axes", min_axes.ToString());
 
            return new RawProto(strName, "", rgChildren);
        }
 
        public static OneHotParameter FromProto(RawProto rp)
        {
            string strVal;
            OneHotParameter p = new OneHotParameter();
 
            if ((strVal = rp.FindValue("axis")) != null)
                p.axis = int.Parse(strVal);
 
            if ((strVal = rp.FindValue("num_output")) != null)
                p.num_output = uint.Parse(strVal);
 
            if ((strVal = rp.FindValue("min")) != null)
                p.min = ParseDouble(strVal);
 
            if ((strVal = rp.FindValue("max")) != null)
                p.max = ParseDouble(strVal);
 
            if ((strVal = rp.FindValue("min_axes")) != null)
                p.min_axes = int.Parse(strVal);
 
            return p;
        }
    }
}