KnnParameter.cs

using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using MyCaffe.basecode;
using System.ComponentModel;
 
namespace MyCaffe.param.beta
{
    [Serializable]
    [TypeConverter(typeof(ExpandableObjectConverter))]
    public class KnnParameter : LayerParameterBase 
    {
        int m_nMaxBatchesToStore = 10;
        int m_nOutput = 10;
        int m_nK = 100;
 
        public KnnParameter()
        {
        }
 
        [Description("Specifies the number of output items (e.g. classes).")]
        public int num_output
        {
            get { return m_nOutput; }
            set { m_nOutput = value; }
        }
 
        [Description("Specifies the number of nearest neighbors to compare per class, selected from items with the shortest distance.  The default = 100, which selects the class with the highest count from the 10 shortest distances.  It is recommended that K > ((2 * num_output) + 1) or convergence may not occur.")]
        public int k
        {
            get { return m_nK; }
            set { m_nK = value; }
        }
 
        [Description("Specifies the maximum number of batches to store and search for neighbors.  Each batch input is stored until the maximum count is reached at which time, the oldest batch is released.  A larger max value = more GPU memory used.")]
        public int max_stored_batches
        {
            get { return m_nMaxBatchesToStore; }
            set { m_nMaxBatchesToStore = value; }
        }
 
        public override object Load(System.IO.BinaryReader br, bool bNewInstance = true)
        {
            RawProto proto = RawProto.Parse(br.ReadString());
            KnnParameter p = FromProto(proto);
 
            if (!bNewInstance)
                Copy(p);
 
            return p;
        }
 
        public override void Copy(LayerParameterBase src)
        {
            KnnParameter p = (KnnParameter)src;
            m_nOutput = p.m_nOutput;
            m_nK = p.m_nK;
            m_nMaxBatchesToStore = p.m_nMaxBatchesToStore;
        }
 
        public override LayerParameterBase Clone()
        {
            KnnParameter p = new KnnParameter();
            p.Copy(this);
            return p;
        }
 
        public override RawProto ToProto(string strName)
        {
            RawProtoCollection rgChildren = new RawProtoCollection();
 
            rgChildren.Add("num_output", num_output.ToString());
            rgChildren.Add("k", k.ToString());
            rgChildren.Add("max_stored_batches", max_stored_batches.ToString());
 
            return new RawProto(strName, "", rgChildren);
        }
 
        public static KnnParameter FromProto(RawProto rp)
        {
            string strVal;
            KnnParameter p = new KnnParameter();
 
            if ((strVal = rp.FindValue("num_output")) != null)
                p.num_output = int.Parse(strVal);
 
            if ((strVal = rp.FindValue("k")) != null)
                p.k = int.Parse(strVal);
 
            if ((strVal = rp.FindValue("max_stored_batches")) != null)
                p.max_stored_batches = int.Parse(strVal);
 
            return p;
        }
    }
}