Pengyu Nie

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

2025 · arXiv (Cornell University)

Machine learning developers frequently use interactive computational notebooks, such as Jupyter notebooks, to host code for data processing and model training. Jupyter notebooks provide a convenient tool for writing machine learning pipelines and interactively observing …