| 141 | } |
| 142 | |
| 143 | int main(int argc, char **argv) { |
| 144 | int device = argc > 1 ? atoi(argv[1]) : 0; |
| 145 | bool console = argc > 2 ? argv[2][0] == '-' : false; |
| 146 | int perc = argc > 3 ? atoi(argv[3]) : 60; |
| 147 | |
| 148 | try { |
| 149 | af::setDevice(device); |
| 150 | af::info(); |
| 151 | naive_bayes_demo(console, perc); |
| 152 | |
| 153 | } catch (af::exception &ae) { std::cerr << ae.what() << std::endl; } |
| 154 | |
| 155 | return 0; |
| 156 | } |
nothing calls this directly
no test coverage detected