提交 c2a7f891 编写于 作者: J jrzaurin

Add commented snippet to illustrate the use of DeepDenseResnet

上级 a3c2b91d
......@@ -181,6 +181,14 @@ deepdense = DeepDense(
embed_input=preprocess_deep.embeddings_input,
continuous_cols=cont_cols,
)
# # To use DeepDenseResnet as the deepdense component simply:
# from pytorch_widedeep.models import DeepDenseResnet:
# deepdense = DeepDenseResnet(
# blocks=[64, 32],
# deep_column_idx=preprocess_deep.deep_column_idx,
# embed_input=preprocess_deep.embeddings_input,
# continuous_cols=cont_cols,
# )
# build, compile and fit
model = WideDeep(wide=wide, deepdense=deepdense)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册