From c2a7f89140b054b9764414fd440a081fe0ef1061 Mon Sep 17 00:00:00 2001
From: jrzaurin <jrzaurin@gmail.com>
Date: Mon, 31 Aug 2020 17:53:41 +0200
Subject: [PATCH] Add commented snippet to illustrate the use of
 DeepDenseResnet

---
 README.md | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/README.md b/README.md
index 87590b5..a5316ee 100644
--- a/README.md
+++ b/README.md
@@ -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)
-- 
GitLab