import numpy as np
import tensorflow as tf

import matplotlib.pyplot as plt
from matplotlib import cm
from mpl_toolkits import mplot3d

from sklearn.datasets import make_moons
from sklearn.model_selection import train_test_split