void OutputUCIInfo(MCTSManager manager, MCTSNode searchRootNode, bool isFinalInfo = false) { BestMoveInfo best = searchRootNode.BestMoveInfo(false); if (numPV == 1) { UCIWriteLine(UCIInfo.UCIInfoString(manager, searchRootNode, best?.BestMoveNode, showWDL: showWDL, scoreAsQ: scoreAsQ)); } else { // Send top move UCIWriteLine(UCIInfo.UCIInfoString(manager, searchRootNode, best.BestMoveNode, 1, showWDL: showWDL, useParentN: !perPVCounters, scoreAsQ: scoreAsQ)); // Send other moves visited MCTSNode[] sortedN = searchRootNode.ChildrenSorted(s => - (float)s.N); int multiPVIndex = 2; for (int i = 0; i < sortedN.Length && i < numPV; i++) { if (!object.ReferenceEquals(sortedN[i], best.BestMoveNode)) { UCIWriteLine(UCIInfo.UCIInfoString(manager, searchRootNode, sortedN[i], multiPVIndex, showWDL: showWDL, useParentN: !perPVCounters, scoreAsQ: scoreAsQ)); multiPVIndex++; } } // Finally show moves that had no visits float elapsedTimeSeconds = (float)(DateTime.Now - manager.StartTimeThisSearch).TotalSeconds; string timeStr = $"{ elapsedTimeSeconds * 1000.0f:F0}"; for (int i = multiPVIndex - 1; i < searchRootNode.NumPolicyMoves; i++) { (MCTSNode node, EncodedMove move, FP16 p)info = searchRootNode.ChildAtIndexInfo(i); if (info.node == null) { bool isWhite = searchRootNode.Annotation.Pos.MiscInfo.SideToMove == SideType.White; EncodedMove moveCorrectPerspective = isWhite ? info.move : info.move.Flipped; string str = $"info depth 0 seldepth 0 time { timeStr } nodes 1 score cp 0 tbhits 0 " + $"multipv {multiPVIndex} pv {moveCorrectPerspective.AlgebraicStr} "; UCIWriteLine(str); multiPVIndex++; } } } if (verboseMoveStats && (logLiveStats || isFinalInfo)) { OutputVerboseMoveStats(CeresEngine.Search.SearchRootNode); } }
public static List <LC0VerboseMoveStat> BuildStats(MCTSNode searchRootNode) { List <LC0VerboseMoveStat> stats = new List <LC0VerboseMoveStat>(); BestMoveInfo best = searchRootNode.BestMoveInfo(false); // First process policy moves not yet expanded // starting from last one (lowest probability). for (int i = searchRootNode.NumPolicyMoves - 1; i >= 0; i--) { (MCTSNode node, EncodedMove move, FP16 p)info = searchRootNode.ChildAtIndexInfo(i); if (info.node == null) { LC0VerboseMoveStat stat = BuildStatNotExpanded(searchRootNode, i); stats.Add(stat); } } // Now process moves expanded in order of visit count. MCTSNode[] sortedN = searchRootNode.ChildrenSorted(s => (float)s.N + 0.0001f * s.P); foreach (MCTSNode node in sortedN) { if (!object.ReferenceEquals(node, best.BestMoveNode)) { stats.Add(BuildStatExpanded(node, false)); } } // Save the best move for last. stats.Add(BuildStatExpanded(best.BestMoveNode, false)); // Finally, output the search root node. stats.Add(BuildStatExpanded(searchRootNode, true)); return(stats); }
/// <summary> /// Writes extensive descriptive information to a specified TextWriter, /// including verbose move statistics, principal variation, and move timing information. /// </summary> /// <param name="searchRootNode"></param> /// <param name="writer"></param> /// <param name="description"></param> public void DumpFullInfo(MGMove bestMove, MCTSNode searchRootNode = null, TextWriter writer = null, string description = null) { searchRootNode = searchRootNode ?? Root; writer = writer ?? Console.Out; int moveIndex = searchRootNode.Tree.Store.Nodes.PriorMoves.Moves.Count; writer.WriteLine(); writer.WriteLine("================================================================================="); writer.Write(DateTime.Now + " SEARCH RESULT INFORMATION, Move = " + ((1 + moveIndex / 2))); writer.WriteLine($" Thread = {Thread.CurrentThread.ManagedThreadId}"); if (description != null) { writer.WriteLine(description); } writer.WriteLine(); writer.WriteLine("Tree root : " + Context.Root); if (searchRootNode != Root) { writer.WriteLine("Search root : " + searchRootNode); } writer.WriteLine(); MCTSNode[] nodesSortedN = null; MCTSNode[] nodesSortedQ = null; string bestMoveInfo = ""; if (searchRootNode.NumChildrenExpanded > 0 && StopStatus != SearchStopStatus.TablebaseImmediateMove && StopStatus != SearchStopStatus.OnlyOneLegalMove) { MCTSNode[] childrenSortedN = searchRootNode.ChildrenSorted(node => - node.N); MCTSNode[] childrenSortedQ = searchRootNode.ChildrenSorted(node => (float)node.Q); bool isTopN = childrenSortedN[0].Annotation.PriorMoveMG == bestMove; bool isTopQ = childrenSortedQ[0].Annotation.PriorMoveMG == bestMove; if (isTopN && isTopQ) { bestMoveInfo = "(TopN and TopQ)"; } else if (isTopN) { bestMoveInfo = "(TopN)"; } else if (isTopQ) { bestMoveInfo = "(TopQ)"; } } // Output position (with history) information. writer.WriteLine("Position : " + searchRootNode.Annotation.Pos.FEN); writer.WriteLine("Tree root position : " + Context.Tree.Store.Nodes.PriorMoves); writer.WriteLine("Search stop status : " + StopStatus); writer.WriteLine("Best move selected : " + bestMove.MoveStr(MGMoveNotationStyle.LC0Coordinate) + " " + bestMoveInfo); writer.WriteLine(); using (new SearchContextExecutionBlock(Context)) { string infoUpdate = UCIInfo.UCIInfoString(this, searchRootNode); writer.WriteLine(infoUpdate); writer.WriteLine(); DumpTimeInfo(writer); writer.WriteLine(); searchRootNode.Dump(1, 1, writer: writer); writer.WriteLine(); MCTSPosTreeNodeDumper.DumpPV(searchRootNode, true, writer); } }
static void DumpNodeStr(PositionWithHistory priorMoves, MCTSNode node, int depth, int countTimesSeen, bool fullDetail) { node.Context.Tree.Annotate(node); char extraChar = ' '; if (node.Terminal == GameResult.Checkmate) { extraChar = 'C'; } else if (node.Terminal == GameResult.Draw) { extraChar = 'D'; } else if (countTimesSeen > 1) { extraChar = countTimesSeen > 9 ? '9' : countTimesSeen.ToString()[0]; } float multiplier = depth % 2 == 0 ? 1.0f : -1.0f; float pctOfVisits = node.IsRoot ? 100.0f : (100.0f * node.N / node.Parent.N); MCTSNode bestMove = null; // TODO: someday show this too MCTSNode nextBestMove = null; if (!node.IsRoot) { MCTSNode[] parentsChildrenSortedQ = node.ChildrenSorted(innerNode => - multiplier * (float)innerNode.Q); if (parentsChildrenSortedQ.Length > 0) { bestMove = parentsChildrenSortedQ[0]; } // if (parentsChildrenSortedQ.Length > 1) nextBestMove = parentsChildrenSortedQ[1]; } // Depth, move Console.Write($"{depth,3}. "); Console.Write(extraChar); Console.Write($" {node.NumPolicyMoves,3} "); Console.Write($"{node.Index,13:N0}"); string san = node.IsRoot ? "" : MGMoveConverter.ToMove(node.Annotation.PriorMoveMG).ToSAN(in node.Parent.Annotation.Pos); // string sanNextBest = node.IsRoot ? "" : MGMoveConverter.ToMove(nextBestMove.Annotation.PriorMoveMG).ToSAN(in node.Parent.Annotation.Pos); if (node.Annotation.Pos.MiscInfo.SideToMove == SideType.White) { Console.Write($" "); Console.Write($"{san,6}"); } else { Console.Write($"{san,6}"); Console.Write($" "); } // float diffBestNextBestQ = 0; // if (nextBestMove != null) diffBestNextBestQ = (float)(bestMove.Q - nextBestMove.Q); // Console.Write($"{ (nextBestMove?.Annotation == null ? "" : nextBestMove.Annotation.PriorMoveMG.ToString()),8}"); // Console.Write($"{diffBestNextBestQ,8:F2}"); Console.Write($"{node.N,13:N0} "); Console.Write($" {pctOfVisits,5:F0}%"); Console.Write($" {100.0 * node.P,6:F2}% "); DumpWithColor(multiplier * node.V, $" {multiplier * node.V,6:F3} ", -0.2f, 0.2f); // DumpWithColor(multiplier * node.VSecondary, $" {multiplier * node.VSecondary,6:F3} ", -0.2f, 0.2f); double q = multiplier * node.Q; DumpWithColor((float)q, $" {q,6:F3} ", -0.2f, 0.2f); // float qStdDev = MathF.Sqrt(node.Ref.VVariance); // if (float.IsNaN(qStdDev)) // Console.WriteLine("found negative var"); // Console.Write($" +/-{qStdDev,5:F2} "); Console.Write($" {node.WinP,5:F2}/{node.DrawP,5:F2}/{node.LossP,5:F2} "); Console.Write($" {node.WAvg,5:F2}/{node.DAvg,5:F2}/{node.LAvg,5:F2} "); // Console.Write($" {node.Ref.QUpdatesWtdAvg,5:F2} "); // Console.Write($" +/-:{MathF.Sqrt(node.Ref.QUpdatesWtdVariance),5:F2} "); // Console.Write($" {node.Ref.TrendBonusToP,5:F2} "); Console.Write($" {node.MPosition,3:F0} "); Console.Write($" {node.MAvg,3:F0} "); if (fullDetail) { int numPieces = node.Annotation.Pos.PieceCount; // Console.Write($" {PosStr(node.Annotation.Pos)} "); Console.Write($" {node.Annotation.Pos.FEN}"); } Console.WriteLine(); }