training_dfd.tex

% Data flow diagram
% Template adapted from http://www.texample.net/tikz/examples/data-flow-diagram/ by David Fokkema
\documentclass{article}
\usepackage{tikz}
\usetikzlibrary{arrows,shapes}

% Custom, should be provided in the same directory.
\usepackage{datastore}

% Disable page number for better SVG.
\pagenumbering{gobble}

\begin{document}

% TODO: \title{ORES machine learning data flow} \maketitle

\begin{tikzpicture}[
  font=\sffamily,
  every matrix/.style={ampersand replacement=\&,column sep=2cm,row sep=2cm},
  interface/.style={draw,thick,regular polygon,regular polygon sides=4,inner sep=0},
  process/.style={draw,thick,rounded corners,inner sep=.3 cm},
  datastore/.style={draw,thick,shape=datastore,inner sep=.3cm},
  to/.style={->,>=stealth',shorten >=1pt,semithick,font=\sffamily\footnotesize},
  every node/.style={align=center}]

  % Position the nodes using a matrix layout
  \matrix{
      \node[interface] (researcher) {Researcher};
        \& \node[interface] (keyer) {Labeler}; \\
	  \node[datastore] (experiments) {Experiments};
        \& \node[process] (wikilabels) {Wiki labels}; \\
	  \node[process] (evaluate) {Evaluate health};
        \& \node[datastore] (training) {Training set};
          \& \node[datastore] (test) {Test set}; \\
        \& \node[process] (train) {Build model}; \\
        \& \node[datastore] (models) {Model}; \\
	  \node[datastore] (results) {Test results};
        \& \node (hidden) {};
		  \& \node[process] (run) {Run model \\ on test set}; \\
  };

  \node [below=1cm, align=flush center] at (hidden)
  {\Large ORES: AI Training \\ \normalsize data flow diagram};

  % Draw the arrows between the nodes and label them.
  \draw[to] (evaluate) to[bend left=50] node[midway, left] {health statistics} (researcher);
  \draw[to] (experiments) -- (wikilabels);
  \draw[to] (researcher) -- node[midway,right] {assumptions \\ choice of data} (experiments);
  \draw[to] (researcher) -- node[midway,left] {choice of model \\ and parameters} (train);
  \draw[to] (wikilabels) to[bend left=25] node[midway,left] {cases} (keyer);
  \draw[to] (keyer) to[bend left=25] node[midway, right] {classification} (wikilabels);
  \draw[to] (wikilabels) -- node[midway,left] {keyed test data} (test);
  \draw[to] (wikilabels) -- (training);
  \draw[to] (results) -- node[midway,left] {feedback} (evaluate);
  \draw[to] (training) -- (train);
  \draw[to] (train) -- (models);
  \draw[to] (models) -- (run);
  \draw[to] (test) -- (run);
  \draw[to] (run) -- node[midway,below] {test scores} (results);
\end{tikzpicture}
\end{document}