Transform flat keys into nested dictionary

Q: Transform flat keys into nested dictionary

This is a Data Manipulation (SQL/Python) interview question from Netflix for Machine Learning Engineer roles. View the full question and solution on PracHub.

Q: How do I approach Data Manipulation (SQL/Python) interview questions?

Data Manipulation (SQL/Python) questions require understanding of core concepts and practice. PracHub provides solutions with explanations to help you master data manipulation (sql/python) interviews.

Question

You are given a flat collection of parameter keys like ['layer1.attention.q_proj.weight', 'layer1.attention.k_proj.weight', 'layer1.mlp.fc1.weight', ...]. Write a Python function that converts this into a nested dictionary grouped by layer index, placing entries that share the same component path under the same layer (e.g., attention, mlp). Define your grouping rule (e.g., strip the 'layer{n}.' prefix to identify similar components), handle nonconforming keys and duplicates, and analyze time and space complexity.

Transform flat keys into nested dictionary

Comments (0)