giuseppenuc/olive-oil-transformer-model
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add register_keras_serializable

Browse Source
This commit is contained in:
Giuseppe Nucifora 2024-12-11 04:19:59 +01:00
parent 32cc73f90f
commit 473419e7c1
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
models/olive_oil/olive_oil-transformer.ipynb
View File

@ -1421,6 +1421,7 @@
" decay_steps=7000 # Aumentato per permettere più esplorazione\n", " decay_steps=7000 # Aumentato per permettere più esplorazione\n",
" )\n", " )\n",
"\n", "\n",
" @keras.saving.register_keras_serializable()\n",
" def weighted_huber_loss(y_true, y_pred):\n", " def weighted_huber_loss(y_true, y_pred):\n",
" # Pesi per diversi output\n", " # Pesi per diversi output\n",
" weights = tf.constant([1.0, 0.8, 0.8, 1.0, 0.6], dtype=tf.float32)\n", " weights = tf.constant([1.0, 0.8, 0.8, 1.0, 0.6], dtype=tf.float32)\n",