diff options
| -rw-r--r-- | training/dtrain/dtrain_net_interface.cc | 2 | ||||
| -rw-r--r-- | training/dtrain/dtrain_net_interface.h | 2 |
2 files changed, 2 insertions, 2 deletions
diff --git a/training/dtrain/dtrain_net_interface.cc b/training/dtrain/dtrain_net_interface.cc index 3b19ecbf..38fad160 100644 --- a/training/dtrain/dtrain_net_interface.cc +++ b/training/dtrain/dtrain_net_interface.cc @@ -218,7 +218,7 @@ main(int argc, char** argv) // debug -- WriteFile f(debug_fn); - *f << debug_output.str(); + f.get() << debug_output.str(); // -- debug // write current weights diff --git a/training/dtrain/dtrain_net_interface.h b/training/dtrain/dtrain_net_interface.h index e603a87f..eb0aa668 100644 --- a/training/dtrain/dtrain_net_interface.h +++ b/training/dtrain/dtrain_net_interface.h @@ -64,7 +64,7 @@ dtrain_net_init(int argc, char** argv, po::variables_map* conf) ("input_weights,w", po::value<string>(), "input weights file") ("learning_rate,l", po::value<weight_t>()->default_value(1.0), "learning rate") ("learning_rate_sparse,l", po::value<weight_t>()->default_value(1.0), "learning rate for sparse features") - ("dense_features,D", po::value<string>()->default_value("EgivenFCoherent SampleCountF CountEF MaxLexFgivenE MaxLexEgivenF IsSingletonF IsSingletonFE Glue WordPenalty PassThrough LanguageModel LanguageModel_OOV"), + ("dense_features,D", po::value<string>()->default_value("EgivenFCoherent SampleCountF CountEF MaxLexFgivenE MaxLexEgivenF IsSingletonF IsSingletonFE Glue WordPenalty PassThrough LanguageModel LanguageModel_OOV Shape_S01111_T11011 Shape_S11110_T11011 Shape_S11100_T11000 Shape_S01110_T01110 Shape_S01111_T01111 Shape_S01100_T11000 Shape_S10000_T10000 Shape_S11100_T11100 Shape_S11110_T11110 Shape_S11110_T11010 Shape_S01100_T11100 Shape_S01000_T01000 Shape_S01010_T01010 Shape_S01111_T01011 Shape_S01100_T01100 Shape_S01110_T11010 Shape_S11000_T11000 Shape_S11000_T01100 IsSupportedOnline ForceRule"), "dense features") ("debug_output,d", po::value<string>()->default_value(""), "file for debug output"); po::options_description cl("Command Line Options"); |