Permalink
Browse files

Added changes from ProtoNN-dev

  • Loading branch information...
adityakusupati committed Nov 23, 2017
1 parent fbc316b commit d7ceeb4ff0255422d1567f7ce803b7657ce72d7f
Showing with 7 additions and 5 deletions.
  1. +4 −3 docs/README_BONSAI_OSS.md
  2. +2 −2 run_BonsaiPredict_usps10.sh
  3. +1 −0 src/Bonsai/BonsaiTrainer.cpp
@@ -104,10 +104,11 @@ For running Training separately followed by prediction
```bash
sh run_BonsaiTrain_usps10.sh
sh run_BonsaiPredict_usps10.sh
The script prints the path of the model-dir
NOTE: You have to change the model_dir param in run_BonsaiPredict_usps10.sh according to the model directory generated by run_BonsaiTrain_usps10.sh - you can find it in <Data-Dir Path>/BonsaiResults/<Model-Dir>
<Data-Dir Path> is generally ./usps10 in this case and <Model-Dir> will have timestamp as its signature(name).
use "ln -s <model-dir> current_model" to set a soft alias(shortcut) if you wish to run on that model or you choose <model-dir> as per your wish so as to use it in BonsaiPredict on usps10
sh run_BonsaiPredict_usps10.sh
```
## Output
@@ -10,7 +10,7 @@
input_dir="-D ./usps10"
input_format="-f 0"
model_dir="-M ./usps10/BonsaiResults/10_21_41_20_11" # Note: The model_dir has to be changed as Model naming is based on timestamp so required to be changed by the user
model_dir="-M current_model" # Note: The model_dir has to be changed as Model naming is based on timestamp so required to be changed by the user
########################################################
# Data-dependent parameters
@@ -25,7 +25,7 @@ ntest="-N 2007"
#gdb=" gdb --args"
executable="./BonsaiPredict"
command=$gdb" "$executable" "$input_format" "$ntest" "$input_dir" "$model_dir
echo "Running Bonsai predict with following command: "
echo "Running Bonsai with following command: "
echo $command
echo ""
exec $command
@@ -631,6 +631,7 @@ void BonsaiTrainer::dumpModelMeanVar(const std::string& currResultsPath)
writeMatrixInASCII(mean, params_path, "Mean");
writeMatrixInASCII(variance, params_path, "Variance");
LOG_INFO(currResultsPath);
}
size_t BonsaiTrainer::totalNonZeros()

0 comments on commit d7ceeb4

Please sign in to comment.