Skip to content

Commit

Permalink
Merge pull request #21 from pronobis/fix/link-einsum-benchmark-and-re…
Browse files Browse the repository at this point in the history
…adme

Fix/link einsum benchmark and readme
  • Loading branch information
jostosh authored Oct 11, 2020
2 parents 6b7508d + 7b28166 commit 0d0a5da
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 2 deletions.
2 changes: 1 addition & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -58,7 +58,7 @@ Currently, the repo is in an alpha state. Hence, one can expect some sporadic br
univariate as well as *multivariate* inputs.

## Examples / Tutorials
1. **Benchmark**: `libspn-keras` andd Einsum Networks. [![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/pronobis/libspn-keras/blob/master/examples/notebooks/Benchmark%20With%20Einsum%20Networks.ipynb)
1. **Benchmark**: `libspn-keras` and Einsum Networks. [![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/pronobis/libspn-keras/blob/master/examples/notebooks/Benchmark%20With%20Einsum%20Networks.ipynb)
2. **Image Classification**: A Deep Generalized Convolutional Sum-Product Network (DGC-SPN). [![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/pronobis/libspn-keras/blob/master/examples/notebooks/DGC-SPN%20Image%20Classification.ipynb)
3. **Image Completion**: A Deep Generalized Convolutional Sum-Product Network (DGC-SPN). [![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/pronobis/libspn-keras/blob/master/examples/notebooks/DGC-SPN%20Image%20Completion.ipynb)
4. **Understanding region SPNs** [![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/pronobis/libspn-keras/blob/master/examples/notebooks/Understanding%20Region%20SPNs.ipynb)
Expand Down
3 changes: 2 additions & 1 deletion examples/notebooks/DGC-SPN Image Completion.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -141,6 +141,7 @@
"source": [
"def define_spn(sum_op, infer_no_evidence=False):\n",
" spnk.set_default_sum_op(sum_op)\n",
" spnk.set_default_linear_accumulator_constraint(spnk.constraints.GreaterEqualEpsilon())\n",
" stack = [\n",
" spnk.layers.NormalizeStandardScore(input_shape=(64, 64, 1)),\n",
" # Non-overlapping products\n",
Expand Down Expand Up @@ -270,7 +271,7 @@
")\n",
"\n",
"sum_product_network.compile(\n",
" optimizer=spnk.optimizers.OnlineExpectationMaximization(), \n",
" optimizer=spnk.optimizers.OnlineExpectationMaximization(learning_rate=1.0, gliding_average=False), \n",
" loss=spnk.losses.NegativeLogLikelihood(),\n",
" metrics=[spnk.metrics.LogLikelihood()]\n",
")\n",
Expand Down

0 comments on commit 0d0a5da

Please sign in to comment.