public class Graph
extends java.lang.Object
Constructor and Description |
---|
Graph() |
Modifier and Type | Method and Description |
---|---|
void |
drawGraph(java.util.ArrayList<java.lang.Double> scores,
boolean save)
Draw a graph of the loss values as a functions of the iterations
|
void |
drawHeatMap(StructEDModel model,
int start,
int jumps)
Draw heat map - used in the OCR tutorial
This function saves the heat map under: img/heat_map.png
|
public void drawGraph(java.util.ArrayList<java.lang.Double> scores, boolean save)
scores
- an array list of the loss valuessave
- a flag indicates whether or not to save the imgpublic void drawHeatMap(StructEDModel model, int start, int jumps) throws java.io.IOException
model
- - the trained modelstart
- - the index from where to start and build the heat mapjumps
- - the row size of the matrixjava.io.IOException