The calculation of the curvature in Volume of Fluid (VOF) methods is still a challenge, and common approaches involve curve or surface fitting based on volume fractions. In this work, we explore an alternative approach for curvature computation in VOF simulations employing machine learning. The neural network establishes a correlation between curvature and height function values so that the local interface curvature can be efficiently predicted. We compare the trained neural network to the standard Height Function method to assess its performance and robustness.