[SPARK-7329] Use itertools.product in ParamGridBuilder

• Type: Improvement
• Status: Resolved
• Priority: Minor
• Resolution: Fixed
• Affects Version/s: 1.4.0
• Fix Version/s: 1.4.0
• Component/s: ML, PySpark
• Target Version/s: 1.4.0
• Assignee: Xiangrui Meng
• Reporter: Xiangrui Meng
• Created: 03/May/15 18:44
• Updated: 04/May/15 01:07
• Resolved: 04/May/15 01:07

Description

justinuang suggested the following on https://github.com/apache/spark/pull/5601:

[dict(zip(self._param_grid.keys(), prod)) for prod in itertools.product(*self._param_grid.values())]

Issue Links

• links to: [Github] Pull Request #5873 (mengxr)