コード例 #1
0
        public FeatureInteractions GetFeatureInteractions(int maxInteractionDepth = -1, int maxDeepening = -1)
        {
            FeatureInteractions xgbFeatureInteractions = new FeatureInteractions();

            _maxInteractionDepth = maxInteractionDepth;
            _maxDeepening        = maxDeepening;

            Console.ResetColor();
            if (_maxInteractionDepth == -1)
            {
                Console.WriteLine(String.Format("Collectiong feature interactions"));
            }
            else
            {
                Console.WriteLine(String.Format("Collectiong feature interactions up to depth {0}", _maxInteractionDepth));
            }

            for (int i = 0; i < NumTrees; i++)
            {
                Console.ForegroundColor = ConsoleColor.DarkGreen;
                Console.Write(String.Format("Collectiong feature interactions within tree #{0} ", i + 1));

                _treeFeatureInteractions = new FeatureInteractions();
                _pathMemo  = new HashSet <string>();
                _treeIndex = i;
                CollectFeatureInteractions(XgbTrees[i], new HashSet <XgbTreeNode>(), currentGain: 0, currentCover: 0, pathProbability: 1, depth: 0, deepening: 0);

                //double treeGain = _treeFeatureInteractions.GetFeatureInteractionsOfDepth(0).Sum(x => x.Value.Gain);
                //foreach (KeyValuePair<string, FeatureInteraction> fi in _treeFeatureInteractions)
                //{
                //    fi.Value.Gain /= treeGain;
                //}

                Console.WriteLine(String.Format("=> number of interactions: {0}", _treeFeatureInteractions.Count));
                Console.ResetColor();
                xgbFeatureInteractions.Merge(_treeFeatureInteractions);
            }

            Console.ForegroundColor = ConsoleColor.Green;
            Console.WriteLine(String.Format("{0} feature interactions has been collected.\n", xgbFeatureInteractions.Count));
            Console.ResetColor();

            return(xgbFeatureInteractions);
        }