| 83 | } |
| 84 | |
| 85 | unsigned predict(ComputationGraph & cg, Instance & inst) { |
| 86 | Expression logits = classify(cg, inst); |
| 87 | std::vector<float> pred = dynet::as_vector(cg.get_value(logits)); |
| 88 | return std::max_element(pred.begin(), pred.end()) - pred.begin(); |
| 89 | } |
| 90 | |
| 91 | unsigned get_max_steps(const Instance & inst) { |
| 92 | unsigned max_steps = 0; |