latex 混淆矩阵

电玩女神 2022-03-12 04:14 411阅读 0赞

我们之前介绍过latex的表格。那么混淆矩阵本质上来讲也是一个表格。怎么简单，快速准确的画出来。

如果先画出来，在插入到论文中，难免会low，而且放缩的话图片会失真。百度上是没有搜索到。所以用google。

首先是不加颜色的，黑白的，使用了Tikz包

\documentclass{article}
    \usepackage[utf8]{inputenc}
    \usepackage[english]{babel}
    
     \usepackage{tikz}
    \usetikzlibrary{positioning}
    
     
    \begin{document}
    \begin{tikzpicture}[
    box/.style={draw,rectangle,minimum size=2cm,text width=1.5cm,align=left}]
    \matrix (conmat) [row sep=.1cm,column sep=.1cm] {
    \node (tpos) [box,
        label=left:\( \mathbf{p'} \),
        label=above:\( \mathbf{p} \),
        ] {True \\ positive};
    &
    \node (fneg) [box,
        label=above:\textbf{n},
        label=above right:\textbf{total},
        label=right:\( \mathrm{P}' \)] {False \\ negative};
    \\
    \node (fpos) [box,
        label=left:\( \mathbf{n'} \),
        label=below left:\textbf{total},
        label=below:P] {False \\ positive};
    &
    \node (tneg) [box,
        label=right:\( \mathrm{N}' \),
        label=below:N] {True \\ negative};
    \\
    };
    \node [left=.05cm of conmat,text width=1.5cm,align=right] {\textbf{actual \\ value}};
    \node [above=.05cm of conmat] {\textbf{prediction outcome}};
    \end{tikzpicture}
    
    \end{document}

长这样：

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3UwMTMyNDk4NTM_size_16_color_FFFFFF_t_70][]

如果是彩色的：

\documentclass[english]{article}
    \usepackage[T1]{fontenc}
    \usepackage[latin9]{inputenc}
    \usepackage{babel}
    \usepackage[table]{xcolor}
    \usepackage{collcell}
    \usepackage{hhline}
    \usepackage{pgf}
    \usepackage{multirow}
    
    \def\colorModel{hsb} %You can use rgb or hsb
    
    \newcommand\ColCell[1]{
      \pgfmathparse{#1<50?1:0}  %Threshold for changing the font color into the cells
        \ifnum\pgfmathresult=0\relax\color{white}\fi
      \pgfmathsetmacro\compA{0}      %Component R or H
      \pgfmathsetmacro\compB{#1/100} %Component G or S
      \pgfmathsetmacro\compC{1}      %Component B or B
      \edef\x{\noexpand\centering\noexpand\cellcolor[\colorModel]{\compA,\compB,\compC}}\x #1
      } 
    \newcolumntype{E}{>{\collectcell\ColCell}m{0.4cm}<{\endcollectcell}}  %Cell width
    \newcommand*\rot{\rotatebox{90}}
    
    \begin{document}
    \newcommand\items{3}   %Number of classes
    \arrayrulecolor{white} %Table line colors
    \noindent\begin{tabular}{cc*{\items}{|E}|}
    \multicolumn{1}{c}{} &\multicolumn{1}{c}{} &\multicolumn{\items}{c}{Predicted} \\ \hhline{~*\items{|-}|}
    \multicolumn{1}{c}{} & 
    \multicolumn{1}{c}{} & 
    \multicolumn{1}{c}{\rot{Class A}} & 
    \multicolumn{1}{c}{\rot{Class B}} & 
    \multicolumn{1}{c}{\rot{Class C}} \\ \hhline{~*\items{|-}|}
    \multirow{\items}{*}{\rotatebox{90}{Actual}} 
    &Class A  & 100   & 0  & 10   \\ \hhline{~*\items{|-}|}
    &Class B  & 10   & 80  & 10   \\ \hhline{~*\items{|-}|}
    &Class C  & 30   & 0   & 70   \\ \hhline{~*\items{|-}|}
    \end{tabular}
    
    \end{document}

![20190305095911846.png][]

改颜色配色：

[https://tex.stackexchange.com/questions/20267/how-to-construct-a-confusion-matrix-in-latex/20284\#20284][https_tex.stackexchange.com_questions_20267_how-to-construct-a-confusion-matrix-in-latex_20284_20284]

还有：

\documentclass{article}
    \usepackage[table]{xcolor}
    \usepackage{pgf}
    \usepackage{tikz}
    \usepackage{collcell}
    \usepackage{caption}
    
    %The min, mid and max values
    \newcommand*{\MinNumber}{0.0}%
    \newcommand*{\MidNumber}{0.2} %
    \newcommand*{\MaxNumber}{1.0}%
    \def\OmitZero#1.#2.#3!{%
      \ifx\relax#2\relax     % #2 is empty -> no decimal
        #1%
      \else                  % #2 not empty -> decimal
        \ifnum#1=0           % we have 0.
          .#2%
        \else                % we have no 0.xx
          #1.#2%
        \fi
      \fi}
    
    %Apply the gradient macro
    \newcommand{\ApplyGradient}[1]{%
        \ifdim #1 pt > \MidNumber pt
        \pgfmathsetmacro{\PercentColor}{max(min(100.0*(#1 - 
        \MidNumber)/(\MaxNumber-\MidNumber),100.0),0.00)} %
        \hspace{-0.33em}\colorbox{green!\PercentColor!red}{\OmitZero#1..!}
        \else
        \pgfmathsetmacro{\PercentColor}{max(min(100.0*(\MidNumber - 
        #1)/(\MidNumber-\MinNumber),100.0),0.00)} %
        \hspace{-0.33em}\colorbox{white!\PercentColor!red}{\OmitZero#1..!}
        \fi
    }
    \newcolumntype{G}{>{\collectcell\ApplyGradient}c<{\endcollectcell}}
    \renewcommand{\arraystretch}{0}
    \setlength{\fboxsep}{3mm} % box size
    \setlength{\tabcolsep}{0pt}
    
    %% for rotation in table
    \usepackage{adjustbox}
    \usepackage{array}
    
    \newcolumntype{R}[2]{%
        >{\adjustbox{angle=#1,lap=\width-(#2)}\bgroup}%
        l%
        <{\egroup}%
    }
    \newcommand*\rotz{\multicolumn{1}{R{0}{-1em}}}% no optional argument here, please! 
    %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
    % END COLORED CM
    %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
    
    \begin{document}
    
    
        \begin{table}[t]
            \caption{The confusion matrix}
            \label{tab:CM}
            \centering
            \resizebox{1\textwidth}{!}{ % to nicely resize the table in the text width
    
                \begin{tabular}{r*{6}{G}}
                    \rotz{} & 
                    \rotz{A} & 
                    \rotz{B} & 
                    \rotz{C} &
                    \rotz{D} & 
                    \rotz{E} \smallskip \\ 
                    A        & 0.54  & 0     & 0.08  & 0.17  & 0     \\ 
    
                    B        & 0     & 0.96  & 0     & 0     & 0     \\ 
    
                    C        & 0.04  & 0.04  & 0.38  & 0.13  & 0     \\ 
    
                    D        & 0.08  & 0     & 0.04  & 0.75  & 0     \\ 
    
                    E        & 0     & 0.04  & 0     & 0     & 0.96  \\ 
    
                \end{tabular}%\par\bigskip
            }
        \end{table}
    
    \end{document}

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3UwMTMyNDk4NTM_size_16_color_FFFFFF_t_70 1][]

实际上，我们比较希望使用：

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3UwMTMyNDk4NTM_size_16_color_FFFFFF_t_70 2][]

那么观察到，这里不是latex的table，而是figure，所以很有可能，是使用python或者matlab画出来然后是加把plot的图片加入到里面了。

首先，利用：[https://stackoverflow.com/questions/2897826/confusion-matrix-with-number-of-classified-misclassified-instances-on-it-python][https_stackoverflow.com_questions_2897826_confusion-matrix-with-number-of-classified-misclassified-instances-on-it-python]

这里的代码生成

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3UwMTMyNDk4NTM_size_16_color_FFFFFF_t_70 3][]

根据[https://ercanozturk.org/2017/12/16/python-matplotlib-plots-in-latex/][https_ercanozturk.org_2017_12_16_python-matplotlib-plots-in-latex]，这里所说。存储成为pdf

然后再添加到latex中:放大后仍旧很清晰。所以，成功了。

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3UwMTMyNDk4NTM_size_16_color_FFFFFF_t_70 4][]

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3UwMTMyNDk4NTM_size_16_color_FFFFFF_t_70]: /images/20220312/51fc3651efbb4f81a42092cee17bd218.png
[20190305095911846.png]: /images/20220312/5ea075cbd80b4ef095ce94a91d7d2b56.png
[https_tex.stackexchange.com_questions_20267_how-to-construct-a-confusion-matrix-in-latex_20284_20284]: https://tex.stackexchange.com/questions/20267/how-to-construct-a-confusion-matrix-in-latex/20284#20284
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3UwMTMyNDk4NTM_size_16_color_FFFFFF_t_70 1]: /images/20220312/64023a7f9ad8419bb6b0a2b425f643f6.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3UwMTMyNDk4NTM_size_16_color_FFFFFF_t_70 2]: /images/20220312/73d9aa54889f405289579614a3c07f41.png
[https_stackoverflow.com_questions_2897826_confusion-matrix-with-number-of-classified-misclassified-instances-on-it-python]: https://stackoverflow.com/questions/2897826/confusion-matrix-with-number-of-classified-misclassified-instances-on-it-python
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3UwMTMyNDk4NTM_size_16_color_FFFFFF_t_70 3]: /images/20220312/d441d828e93c4c86be40380817c0ec28.png
[https_ercanozturk.org_2017_12_16_python-matplotlib-plots-in-latex]: https://ercanozturk.org/2017/12/16/python-matplotlib-plots-in-latex/
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3UwMTMyNDk4NTM_size_16_color_FFFFFF_t_70 4]: /images/20220312/8155aadcb6bc44e1ae29137bf6e9d689.png