Esempio n. 1
0
        private void TryAddSupplementalNodes(MCTSManager manager, int maxNodes,
                                             MCTSNodesSelectedSet selectedNodes, ILeafSelector selector)
        {
            foreach ((MCTSNode parentNode, int selectorID, int childIndex) in ((LeafSelectorMulti)selector).supplementalCandidates) // TODO: remove cast
            {
                if (childIndex <= parentNode.NumChildrenExpanded - 1)
                {
                    // This child was already selected as part of the normal leaf gathering process.
                    continue;
                }
                else
                {
                    MCTSEventSource.TestCounter1++;

                    // Record visit to this child in the parent (also increments the child NInFlight counter)
                    parentNode.UpdateRecordVisitsToChild(selectorID, childIndex, 1);

                    MCTSNode node = parentNode.CreateChild(childIndex);

                    ((LeafSelectorMulti)selector).DoVisitLeafNode(node, 1);// TODO: remove cast

                    if (!parentNode.IsRoot)
                    {
                        if (selectorID == 0)
                        {
                            parentNode.Parent.Ref.BackupIncrementInFlight(1, 0);
                        }
                        else
                        {
                            parentNode.Parent.Ref.BackupIncrementInFlight(0, 1);
                        }
                    }

                    // Try to process this node
                    int nodesBefore = selectedNodes.NodesNN.Count;
                    selector.InsureAnnotated(node);
                    selectedNodes.ProcessNode(node);
                    bool wasSentToNN = selectedNodes.NodesNN.Count != nodesBefore;
                    //if (wasSentToNN) MCTSEventSource.TestCounter2++;

                    // dje: add counter?
                }
            }
        }
Esempio n. 2
0
        private void TryAddRootPreloadNodes(MCTSManager manager, int maxNodes, MCTSNodesSelectedSet selectedNodes, ILeafSelector selector)
        {
            if (rootPreloader == null)
            {
                return;
            }

            List <MCTSNode> rootPreloadNodes = rootPreloader.GetRootPreloadNodes(manager.Root, selector.SelectorID, maxNodes, MCTSRootPreloader.PRELOAD_MIN_P);

            if (rootPreloadNodes != null)
            {
                for (int i = 0; i < rootPreloadNodes.Count; i++)
                {
                    MCTSNode node = rootPreloadNodes[i];
                    selector.InsureAnnotated(node);
                    selectedNodes.ProcessNode(node);
                }
            }
        }