提交 8a283dbc 编写于 作者: W wangmeng28

Update docs for fm layer

上级 89e63b13
...@@ -7423,18 +7423,25 @@ def factorization_machine(input, ...@@ -7423,18 +7423,25 @@ def factorization_machine(input,
Factorization machines. Factorization machines.
.. code-block:: python .. code-block:: python
factor_machine = factorization_machine(input=input_layer, factor_size=10) first_order = paddle.layer.fc(input=input,
size=1,
:param input: The input layer. 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 :type input: LayerOutput
:param factor_size: The hyperparameter that defines the dimensionality of :param factor_size: The hyperparameter that defines the dimensionality of
the latent vector size the latent vector size.
:type context_len: int :type context_len: int
:param act: Activation Type. Default is linear activation. :param act: Activation Type. Default is linear activation.
:type act: BaseActivation :type act: BaseActivation
:param param_attr: The Parameter Attribute. If None, the latent vectors will :param param_attr: The parameter attribute. See ParameterAttribute for
be initialized smartly. It's better to set it by details.
yourself.
:type param_attr: ParameterAttribute :type param_attr: ParameterAttribute
:param layer_attr: Extra Layer config. :param layer_attr: Extra Layer config.
:type layer_attr: ExtraLayerAttribute|None :type layer_attr: ExtraLayerAttribute|None
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册