EST_Wagon.h (speech_tools-2.4-release) | : | EST_Wagon.h (speech_tools-2.5.0-release) | ||
---|---|---|---|---|
skipping to change at line 50 | skipping to change at line 50 | |||
#ifndef __WAGON_H__ | #ifndef __WAGON_H__ | |||
#define __WAGON_H__ | #define __WAGON_H__ | |||
#include "EST_String.h" | #include "EST_String.h" | |||
#include "EST_Val.h" | #include "EST_Val.h" | |||
#include "EST_TVector.h" | #include "EST_TVector.h" | |||
#include "EST_TList.h" | #include "EST_TList.h" | |||
#include "EST_simplestats.h" /* For EST_SuffStats class */ | #include "EST_simplestats.h" /* For EST_SuffStats class */ | |||
#include "EST_Track.h" | #include "EST_Track.h" | |||
#include "siod.h" | #include "siod.h" | |||
// When set to one wagon supports using multiple threads if | ||||
// --omp_nthreads X is used (works for most gccs) | ||||
// #define OMP_WAGON 1 | ||||
#ifdef OMP_WAGON | ||||
#include "omp.h" | ||||
#endif | ||||
#define wagon_error(WMESS) (cerr << WMESS << endl,exit(-1)) | #define wagon_error(WMESS) (cerr << WMESS << endl,exit(-1)) | |||
// I get floating point exceptions of Alphas when I do any comparisons | // I get floating point exceptions of Alphas when I do any comparisons | |||
// with HUGE_VAL or FLT_MAX so I'll make my own | // with HUGE_VAL or FLT_MAX so I'll make my own | |||
#define WGN_HUGE_VAL 1.0e20 | #define WGN_HUGE_VAL 1.0e20 | |||
class WVector : public EST_FVector | class WVector : public EST_FVector | |||
{ | { | |||
public: | public: | |||
WVector(int n) : EST_FVector(n) {} | WVector(int n) : EST_FVector(n) {} | |||
skipping to change at line 261 | skipping to change at line 269 | |||
void wgn_load_dataset(WDataSet &ds,EST_String fname); | void wgn_load_dataset(WDataSet &ds,EST_String fname); | |||
WNode *wgn_build_tree(float &score); | WNode *wgn_build_tree(float &score); | |||
WNode *wgn_build_dlist(float &score,ostream *output); | WNode *wgn_build_dlist(float &score,ostream *output); | |||
WNode *wagon_stepwise(float limit); | WNode *wagon_stepwise(float limit); | |||
float wgn_score_question(WQuestion &q, WVectorVector &ds); | float wgn_score_question(WQuestion &q, WVectorVector &ds); | |||
void wgn_find_split(WQuestion &q,WVectorVector &ds, | void wgn_find_split(WQuestion &q,WVectorVector &ds, | |||
WVectorVector &y,WVectorVector &n); | WVectorVector &y,WVectorVector &n); | |||
float summary_results(WNode &tree,ostream *output); | float summary_results(WNode &tree,ostream *output); | |||
extern int wgn_min_cluster_size; | extern int wgn_min_cluster_size; | |||
extern int wgn_max_questions; | ||||
extern int wgn_held_out; | extern int wgn_held_out; | |||
extern float wgn_dropout_feats; | ||||
extern float wgn_dropout_samples; | ||||
extern int wgn_cos; | ||||
extern int wgn_prune; | extern int wgn_prune; | |||
extern int wgn_quiet; | extern int wgn_quiet; | |||
extern int wgn_verbose; | extern int wgn_verbose; | |||
extern int wgn_predictee; | extern int wgn_predictee; | |||
extern int wgn_count_field; | extern int wgn_count_field; | |||
extern EST_String wgn_count_field_name; | extern EST_String wgn_count_field_name; | |||
extern EST_String wgn_predictee_name; | extern EST_String wgn_predictee_name; | |||
extern float wgn_float_range_split; | extern float wgn_float_range_split; | |||
extern float wgn_balance; | extern float wgn_balance; | |||
extern EST_String wgn_opt_param; | extern EST_String wgn_opt_param; | |||
End of changes. 3 change blocks. | ||||
0 lines changed or deleted | 12 lines changed or added |