| 18 | using namespace af; |
| 19 | |
| 20 | float accuracy(const array &predicted, const array &target) { |
| 21 | array val, plabels, tlabels; |
| 22 | max(val, tlabels, target, 1); |
| 23 | max(val, plabels, predicted, 1); |
| 24 | |
| 25 | return 100 * count<float>(plabels == tlabels) / tlabels.elements(); |
| 26 | } |
| 27 | |
| 28 | // Predict based on given parameters |
| 29 | array predict(const array &X, const array &Weights) { |
no test coverage detected