Yuchan commited on
Commit
b88baf1
·
verified ·
1 Parent(s): b890427

Update Mo.py

Browse files
Files changed (1) hide show
  1. Mo.py +1 -8
Mo.py CHANGED
@@ -226,13 +226,6 @@ def smoothed_loss_keras(y_true, y_pred, eps=0.1):
226
  per_tok = per_tok * mask
227
  return tf.reduce_sum(per_tok) / (tf.reduce_sum(mask) + 1e-8)
228
 
229
- def masked_accuracy(y_true, y_pred):
230
- y_true = tf.cast(y_true, tf.int32)
231
- mask = tf.cast(tf.not_equal(y_true, pad_id), tf.float32)
232
- pred_id = tf.argmax(y_pred, axis=-1, output_type=tf.int32)
233
- acc = tf.cast(tf.equal(y_true, pred_id), tf.float32) * mask
234
- return tf.reduce_sum(acc) / (tf.reduce_sum(mask) + 1e-8)
235
-
236
  def masked_perplexity(y_true, y_pred, eps=0.1):
237
  y_true = tf.cast(y_true, tf.int32)
238
  mask = tf.cast(tf.not_equal(y_true, pad_id), tf.float32)
@@ -255,7 +248,7 @@ with strategy.scope():
255
  model.summary()
256
 
257
  optimizer = tf.keras.optimizers.Adam(1e-4, beta_1=0.9, beta_2=0.95, epsilon=1e-8, clipnorm=1.0)
258
- model.compile(optimizer=optimizer, loss=smoothed_loss_keras, metrics=[masked_accuracy, masked_perplexity])
259
 
260
  # 학습
261
  history = model.fit(dist_dataset, epochs=1, verbose=1)
 
226
  per_tok = per_tok * mask
227
  return tf.reduce_sum(per_tok) / (tf.reduce_sum(mask) + 1e-8)
228
 
 
 
 
 
 
 
 
229
  def masked_perplexity(y_true, y_pred, eps=0.1):
230
  y_true = tf.cast(y_true, tf.int32)
231
  mask = tf.cast(tf.not_equal(y_true, pad_id), tf.float32)
 
248
  model.summary()
249
 
250
  optimizer = tf.keras.optimizers.Adam(1e-4, beta_1=0.9, beta_2=0.95, epsilon=1e-8, clipnorm=1.0)
251
+ model.compile(optimizer=optimizer, loss=smoothed_loss_keras, metrics=[masked_perplexity])
252
 
253
  # 학습
254
  history = model.fit(dist_dataset, epochs=1, verbose=1)