Extract dimensions in evaluation
Test on toy data
Merge