diff --git a/sample/main.cc b/sample/main.cc index 5d11d9a..8d65ba0 100644 --- a/sample/main.cc +++ b/sample/main.cc @@ -226,6 +226,7 @@ int main(int argc, char** argv) //showCPDS(network); cout << "Score: " << network.score(Xd, y) << endl; cout << "PyTorch version: " << TORCH_VERSION << endl; + cout << "BayesNet version: " << network.version() << endl; unsigned int nthreads = std::thread::hardware_concurrency(); cout << "Computer has " << nthreads << " cores." << endl; return 0; diff --git a/src/Network.cc b/src/Network.cc index 130d317..458eadb 100644 --- a/src/Network.cc +++ b/src/Network.cc @@ -100,7 +100,6 @@ namespace bayesnet { if (maxThreadsRunning < 1) { maxThreadsRunning = 1; } - cout << "Using " << maxThreadsRunning << " threads" << " maxThreads: " << maxThreads << endl; vector threads; mutex mtx; condition_variable cv; diff --git a/src/Network.h b/src/Network.h index 2eb2b82..db4a1e3 100644 --- a/src/Network.h +++ b/src/Network.h @@ -36,6 +36,7 @@ namespace bayesnet { vector predict(const vector>&); vector> predict_proba(const vector>&); double score(const vector>&, const vector&); + inline string version() { return "0.1.0"; } }; } #endif \ No newline at end of file