public static final class Word2VecV3.Word2VecParametersV3 extends water.api.schemas3.ModelParametersSchemaV3<Word2VecModel.Word2VecParameters,Word2VecV3.Word2VecParametersV3>
Modifier and Type | Field and Description |
---|---|
int |
epochs |
static java.lang.String[] |
fields |
float |
init_learning_rate |
int |
min_word_freq |
Word2Vec.NormModel |
norm_model |
water.api.schemas3.KeyV3.FrameKeyV3 |
pre_trained |
float |
sent_sample_rate |
int |
vec_size |
int |
window_size |
Word2Vec.WordModel |
word_model |
auc_type, categorical_encoding, checkpoint, custom_distribution_func, custom_metric_func, distribution, export_checkpoints_dir, fold_assignment, fold_column, gainslift_bins, huber_alpha, ignore_const_cols, ignored_columns, keep_cross_validation_fold_assignment, keep_cross_validation_models, keep_cross_validation_predictions, max_categorical_levels, max_runtime_secs, model_id, nfolds, offset_column, parallelize_cross_validation, quantile_alpha, response_column, score_each_iteration, stopping_metric, stopping_rounds, stopping_tolerance, training_frame, tweedie_power, validation_frame, weights_column
Constructor and Description |
---|
Word2VecParametersV3() |
append_field_arrays, extractDeclaredApiParameters, fields, fillFromImpl, fillImpl, getAdditionalParameters, writeParametersJSON
createAndFillImpl, createImpl, extractVersionFromSchemaName, fillFromAny, fillFromBody, fillFromImpl, fillFromImpl, fillFromParms, fillFromParms, fillFromParms, fillImpl, getImplClass, getImplClass, getSchemaName, getSchemaType, getSchemaVersion, init_meta, markdown, markdown, newInstance, newInstance, setField, setSchemaType_doNotCall
public static java.lang.String[] fields
@API(help="Set size of word vectors") public int vec_size
@API(help="Set max skip length between words") public int window_size
@API(help="Set threshold for occurrence of words. Those that appear with higher frequency in the training data\n\t\twill be randomly down-sampled; useful range is (0, 1e-5)") public float sent_sample_rate
@API(help="Use Hierarchical Softmax", values="HSM") public Word2Vec.NormModel norm_model
@API(help="Number of training iterations to run") public int epochs
@API(help="This will discard words that appear less than <int> times") public int min_word_freq
@API(help="Set the starting learning rate") public float init_learning_rate
@API(help="The word model to use (SkipGram or CBOW)", values={"SkipGram","CBOW"}) public Word2Vec.WordModel word_model
@API(help="Id of a data frame that contains a pre-trained (external) word2vec model") public water.api.schemas3.KeyV3.FrameKeyV3 pre_trained