Approximation with Neural Network not working, predicts a constant value throughout the domain

Question

I\'m writing a script to infer a plane from 2D data. So far, this is what I came up with:

import matplotlib.pyplot as plt
import numpy as np


def interp(data