Skip to content

Instantly share code, notes, and snippets.

@sadimanna
Created July 29, 2020 15:03
Show Gist options
  • Select an option

  • Save sadimanna/b6cef68d62925ffad018a14a87c5da37 to your computer and use it in GitHub Desktop.

Select an option

Save sadimanna/b6cef68d62925ffad018a14a87c5da37 to your computer and use it in GitHub Desktop.
def loss_fn(y_true,y_pred):
loss = 0
loss -= (positive_weights['desert']*y_true[0]*K.log(y_pred[0]) + negative_weights['desert']*(1-y_true[0])*K.log(1-y_pred[0]))
loss -= (positive_weights['mountains']*y_true[1]*K.log(y_pred[1]) + negative_weights['mountains']*(1-y_true[1])*K.log(1-y_pred[1]))
loss -= (positive_weights['sea']*y_true[2]*K.log(y_pred[2]) + negative_weights['sea']*(1-y_true[2])*K.log(1-y_pred[2]))
loss -= (positive_weights['sunset']*y_true[3]*K.log(y_pred[3]) + negative_weights['sunset']*(1-y_true[3])*K.log(1-y_pred[3]))
loss -= (positive_weights['trees']*y_true[4]*K.log(y_pred[4]) + negative_weights['trees']*(1-y_true[4])*K.log(1-y_pred[4]))
return loss
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment