public static void main(String[] args){ Model model = makeReminderModel(); Dataset dataset = Dataset.fromText("data/examples-reminders.txt"); HParams hparams = HParams.hparams() .withLearnRate(0.08f) .withL2Penalty(0.01f) .set(SVMOptimizer.CORRECT_PROB, 0.4f); Experiment experiment = new Experiment(model, dataset, hparams, "model/reminders.weights"); // Uncomment to train the model //experiment.train(120, true); experiment.evaluate(); // Uncomment to run model with user input //experiment.interactive(); } }
@Test void saveLoad() { HParams hparams = HParams.hparams() .withLearnRate(0.08f) .withL2Penalty(0.01f) .set(SVMOptimizer.CORRECT_PROB, 0.4f); Dataset d = Dataset.fromText("data/examples-reminders.txt"); Model m = makeReminderModel(); Optimizer optimizer = new SVMOptimizer(m, hparams); m.train(d, optimizer,30); float acc = m.evaluate(d, 0); m.saveWeights("test-weights"); m = makeReminderModel(); m.loadWeights("test-weights"); float savedAcc = m.evaluate(d, 0); assertEquals(acc, savedAcc, 1e-7); } }