Count rows per groupDeep-ML | Problem