Suggesting Code Edits in Interactive Machine Learning Notebooks Using Large Language Models

Bihui Jin, Jiayue Wang, Pengyu Nie

Machine learning developers frequently use interactive computational notebooks, such as Jupyter notebooks, to host code for data processing and model training. Jupyter notebooks pr…