In [0]:

    

import numpy as np
from qutip import *
import networkx as nx
from matplotlib import pyplot as plt
import tensorflow as tf
from NN_learn import train_NN_ising
import os, shutil

from tensorflow.keras.layers import Activation
from tensorflow.keras import backend as K
from tensorflow.keras.utils import get_custom_objects
from tensorflow.keras import layers
from tensorflow import keras
def swish(x):
    return x * K.sigmoid(x)

get_custom_objects().update({'swish': Activation(swish)})