Machine learning aided finite element non-uniform mesh generation