gnes.encoder.text.w2v module¶

class gnes.encoder.text.w2v.Word2VecEncoder(model_dir, skiprows=1, dimension=300, pooling_strategy='REDUCE_MEAN', *args, **kwargs)[source]¶

Bases: gnes.encoder.base.BaseTextEncoder

encode(text, *args, **kwargs)[source]¶
Return type:ndarray
is_trained = True¶
post_init()[source]¶

Declare class attributes/members that can not be serialized in standard way

train(*args, **kwargs)¶

Train the model, need to be overrided