Problem · Array
Implement Cross-Entropy Loss
Problem statement
You are given a batch of model probability distributions probabilities and the correct class index for each example in labels.
Return the mean multiclass cross-entropy loss:
-(1 / n) * sum(log(probabilities[i][labels[i]])).
The problem statement continues
ProExamples
Example 1
probabilities = [[0.7,0.2,0.1],[0.1,0.5,0.4]]labels = [0,2]return = 0.6364828379The mean loss is (-log(0.7) - log(0.4)) / 2.
FastPrep Pro
Reported in 1 OpenAI interview this weekUnlock this recently reported problem
FastPrep Pro gives you full access to interview problems reported within the last week.
- Full problem statement and constraints
- 1 more worked example, explained
- Guided hints and editorial
- Run your code on real test cases
$8.25/month
$99 billed yearly — or $19 month-to-month. Cancel anytime.
Free plan — 2 of 2 free unlocks used this week