tfl.premade_lib.build_linear_layer
Stay organized with collections
Save and categorize content based on your preferences.
Creates a tfl.layers.Linear layer initialized to be an average.
tfl.premade_lib.build_linear_layer(
linear_input,
feature_configs,
model_config,
weighted_average,
submodel_index,
dtype
)
Args |
|---|
linear_input
|
Input to the linear layer.
|
feature_configs
|
A list of tfl.configs.FeatureConfig instances that
specify configurations for each feature.
|
model_config
|
Model configuration object describing model architecture.
Should be one of the model configs in tfl.configs.
|
weighted_average
|
If the linear coefficients should be positive and sum up
to one.
|
submodel_index
|
Corresponding index into submodels.
|
dtype
|
dtype
|
Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates.
Last updated 2024-08-02 UTC.
[null,null,["Last updated 2024-08-02 UTC."],[],[],null,["# tfl.premade_lib.build_linear_layer\n\n\u003cbr /\u003e\n\n|-------------------------------------------------------------------------------------------------------------------------------|\n| [View source on GitHub](https://github.com/tensorflow/lattice/blob/v2.1.1/tensorflow_lattice/python/premade_lib.py#L482-L532) |\n\nCreates a [`tfl.layers.Linear`](../../tfl/layers/Linear) layer initialized to be an average.\n\n#### View aliases\n\n\n**Main aliases**\n\n[`tfl.premade.premade_lib.build_linear_layer`](https://www.tensorflow.org/lattice/api_docs/python/tfl/premade_lib/build_linear_layer)\n\n\u003cbr /\u003e\n\n tfl.premade_lib.build_linear_layer(\n linear_input,\n feature_configs,\n model_config,\n weighted_average,\n submodel_index,\n dtype\n )\n\n\u003cbr /\u003e\n\n\u003cbr /\u003e\n\n\u003cbr /\u003e\n\n| Args ---- ||\n|--------------------|-------------------------------------------------------------------------------------------------------------------------------------|\n| `linear_input` | Input to the linear layer. |\n| `feature_configs` | A list of [`tfl.configs.FeatureConfig`](../../tfl/configs/FeatureConfig) instances that specify configurations for each feature. |\n| `model_config` | Model configuration object describing model architecture. Should be one of the model configs in [`tfl.configs`](../../tfl/configs). |\n| `weighted_average` | If the linear coefficients should be positive and sum up to one. |\n| `submodel_index` | Corresponding index into submodels. |\n| `dtype` | dtype |\n\n\u003cbr /\u003e\n\n\u003cbr /\u003e\n\n\u003cbr /\u003e\n\n\u003cbr /\u003e\n\n| Returns ------- ||\n|---|---|\n| A [`tfl.layers.Linear`](../../tfl/layers/Linear) instance. ||\n\n\u003cbr /\u003e"]]
View source on GitHub