From 8a283dbc9e78f8c2f00d04180986abfb7d6b29df Mon Sep 17 00:00:00 2001 From: wangmeng28 Date: Mon, 27 Nov 2017 19:13:28 +0800 Subject: [PATCH] Update docs for fm layer --- .../paddle/trainer_config_helpers/layers.py | 21 ++++++++++++------- 1 file changed, 14 insertions(+), 7 deletions(-) diff --git a/python/paddle/trainer_config_helpers/layers.py b/python/paddle/trainer_config_helpers/layers.py index 32287cce6cc..288aebb5b49 100644 --- a/python/paddle/trainer_config_helpers/layers.py +++ b/python/paddle/trainer_config_helpers/layers.py @@ -7423,18 +7423,25 @@ def factorization_machine(input, Factorization machines. .. code-block:: python - factor_machine = factorization_machine(input=input_layer, factor_size=10) - - :param input: The input layer. + first_order = paddle.layer.fc(input=input, + size=1, + act=paddle.activation.Linear()) + second_order = paddle.layer.factorization_machine(input=input, + factor_size=10) + fm = paddle.layer.addto(input=[first_order, second_order], + act=paddle.activation.Linear(), + bias_attr=False) + + :param input: The input layer. Supported input types: all input data types + on CPU, and only dense input types on GPU. :type input: LayerOutput :param factor_size: The hyperparameter that defines the dimensionality of - the latent vector size + the latent vector size. :type context_len: int :param act: Activation Type. Default is linear activation. :type act: BaseActivation - :param param_attr: The Parameter Attribute. If None, the latent vectors will - be initialized smartly. It's better to set it by - yourself. + :param param_attr: The parameter attribute. See ParameterAttribute for + details. :type param_attr: ParameterAttribute :param layer_attr: Extra Layer config. :type layer_attr: ExtraLayerAttribute|None -- GitLab