NeuralNetProgressPage.java

package water.api;

import hex.NeuralNet;
import water.Job;
import water.Key;
import water.UKV;

public class NeuralNetProgressPage extends Progress2 {
  /** Return {@link Response} for finished job. */
  @Override protected Response jobDone(final Key dst) {
    return NeuralNetModelView.redirect(this, dst);
  }

  public static Response redirect(Request req, Key jobkey, Key dest) {
    return Response.redirect(req, "/2/NeuralNetProgressPage", JOB_KEY, jobkey, DEST_KEY, dest);
  }

  @Override public boolean toHTML( StringBuilder sb ) {
    Job jjob = Job.findJob(job_key);
    if (jjob ==null) return true;
    NeuralNet.NeuralNetModel m = UKV.get(jjob.dest());
    if (m!=null) m.generateHTML("NeuralNet Model", sb);
    else DocGen.HTML.paragraph(sb, "Pending...");
    return true;
  }
}