def test_save_load(): """Test the save/load operations of PGNN""" PhysicsGuidedNeuralNetwork.seed(0) model = PhysicsGuidedNeuralNetwork(p_fun=p_fun_pythag, hidden_layers=HIDDEN_LAYERS, loss_weights=(0.0, 1.0), n_features=2, n_labels=1, feature_names=['a', 'b'], output_names=['c']) model.fit(X, Y_NOISE, P, n_batch=4, n_epoch=20) y_pred = model.predict(X) with tempfile.TemporaryDirectory() as td: fpath = os.path.join(td, 'tempfile.pkl') model.save(fpath) loaded = PhysicsGuidedNeuralNetwork.load(fpath) assert len(model.layers) == len(loaded.layers) for layer0, layer1 in zip(model.layers, loaded.layers): for i in range(len(layer0.weights)): assert layer0.weights[i].shape == layer1.weights[i].shape assert np.allclose(layer0.weights[i], layer1.weights[i]) y_pred_loaded = loaded.predict(X) assert np.allclose(y_pred, y_pred_loaded) assert loaded.feature_names == ['a', 'b'] assert loaded.output_names == ['c'] assert isinstance(model._optimizer, Adam) assert isinstance(loaded._optimizer, Adam) assert model._optimizer.get_config() == loaded._optimizer.get_config()
def test_conv3d(): """Test a phygnn model with a conv3d layer. The data in this test is garbage, just a test on shapes and save/load functionality""" input_layer = { 'class': 'Conv3D', 'filters': 2, 'kernel_size': 3, 'activation': 'relu' } hidden_layers = [{'units': 64, 'activation': 'relu'}, {'class': 'Flatten'}] output_layer = {'units': 24} model = PhysicsGuidedNeuralNetwork(p_fun=p_fun_pythag, hidden_layers=hidden_layers, input_layer=input_layer, output_layer=output_layer, loss_weights=(1.0, 0.0), n_features=1, n_labels=24) train_x_bad = np.random.uniform(-1, 1, (50, 12, 7, 7, 2)) train_x = np.random.uniform(-1, 1, (50, 12, 7, 7, 1)) train_y = np.random.uniform(-1, 1, (50, 24)) assert len(model.layers) == 5, "conv layers did not get added!" assert isinstance(model.layers[0], Conv3D) assert isinstance(model.layers[1], Dense) assert isinstance(model.layers[2], Activation) assert isinstance(model.layers[3], Flatten) assert isinstance(model.layers[4], Dense) # test raise on bad feature channel dimension with pytest.raises(AssertionError): model.fit(train_x_bad, train_y, train_x, n_batch=1, n_epoch=10) model.fit(train_x, train_y, train_x, n_batch=1, n_epoch=10) y_pred = model.predict(train_x) assert y_pred.shape == (50, 24) with tempfile.TemporaryDirectory() as td: fpath = os.path.join(td, 'tempfile.pkl') model.save(fpath) loaded = PhysicsGuidedNeuralNetwork.load(fpath) assert len(model.layers) == len(loaded.layers) for layer0, layer1 in zip(model.layers, loaded.layers): for i in range(len(layer0.weights)): assert layer0.weights[i].shape == layer1.weights[i].shape assert np.allclose(layer0.weights[i], layer1.weights[i]) y_pred_loaded = loaded.predict(train_x) assert np.allclose(y_pred, y_pred_loaded) assert len(model.layers) == len(loaded.layers)
def test_batch_norm(): """Test the addition of BatchNormalization layers""" HIDDEN_LAYERS = [ { 'units': 64 }, { 'batch_normalization': { 'axis': 1 } }, { 'activation': 'relu' }, { 'units': 64, 'activation': 'relu', 'batch_normalization': { 'axis': 1 } }, ] model = PhysicsGuidedNeuralNetwork(p_fun=p_fun_pythag, hidden_layers=HIDDEN_LAYERS, loss_weights=(0.0, 1.0), n_features=2, n_labels=1) assert len(model.layers) == 8, "Batch norm layers did not get added!" assert isinstance(model.layers[0], InputLayer) assert isinstance(model.layers[1], Dense) assert isinstance(model.layers[2], BatchNormalization) assert isinstance(model.layers[3], Activation) assert isinstance(model.layers[4], Dense) assert isinstance(model.layers[5], BatchNormalization) assert isinstance(model.layers[6], Activation) model.fit(X, Y_NOISE, P, n_batch=1, n_epoch=10) y_pred = model.predict(X) with tempfile.TemporaryDirectory() as td: fpath = os.path.join(td, 'tempfile.pkl') model.save(fpath) loaded = PhysicsGuidedNeuralNetwork.load(fpath) y_pred_loaded = loaded.predict(X) assert np.allclose(y_pred, y_pred_loaded) assert len(model.layers) == len(loaded.layers)
def test_lstm(): """Test a phygnn model with a conv1d layer. The data in this test is garbage, just a test on shapes and creation. Save/load doesnt work yet for lstm""" input_layer = {'class': 'LSTM', 'units': 24, 'return_sequences': True} hidden_layers = [{'units': 64, 'activation': 'relu'}] output_layer = {'units': 24} model = PhysicsGuidedNeuralNetwork(p_fun=p_fun_pythag, hidden_layers=hidden_layers, input_layer=input_layer, output_layer=output_layer, loss_weights=(1.0, 0.0), n_features=2, n_labels=24) train_x = np.random.uniform(-1, 1, (50, 12, 2)) train_y = np.random.uniform(-1, 1, (50, 12, 24)) assert len(model.layers) == 4, "lstm layers did not get added!" assert isinstance(model.layers[0], LSTM) assert isinstance(model.layers[1], Dense) assert isinstance(model.layers[2], Activation) assert isinstance(model.layers[3], Dense) model.fit(train_x, train_y, train_x, n_batch=1, n_epoch=10) y_pred = model.predict(train_x) assert y_pred.shape == (50, 12, 24) with tempfile.TemporaryDirectory() as td: fpath = os.path.join(td, 'tempfile.pkl') model.save(fpath) loaded = PhysicsGuidedNeuralNetwork.load(fpath) assert len(model.layers) == len(loaded.layers) for layer0, layer1 in zip(model.layers, loaded.layers): for i in range(len(layer0.weights)): assert layer0.weights[i].shape == layer1.weights[i].shape assert np.allclose(layer0.weights[i], layer1.weights[i]) y_pred_loaded = loaded.predict(train_x) assert np.allclose(y_pred, y_pred_loaded) assert len(model.layers) == len(loaded.layers)
def test_save_load(): """Test the save/load operations of PGNN""" PhysicsGuidedNeuralNetwork.seed(0) model = PhysicsGuidedNeuralNetwork(p_fun=p_fun_pythag, hidden_layers=HIDDEN_LAYERS, loss_weights=(0.0, 1.0), n_features=2, n_labels=1, feature_names=['a', 'b'], output_names=['c']) model.fit(X, Y_NOISE, P, n_batch=4, n_epoch=20) y_pred = model.predict(X) model.save(FPATH) loaded = PhysicsGuidedNeuralNetwork.load(FPATH) y_pred_loaded = loaded.predict(X) assert np.allclose(y_pred, y_pred_loaded) assert loaded.feature_names == ['a', 'b'] assert loaded.output_names == ['c'] os.remove(FPATH)
def test_dropouts(): """Test the dropout rate kwargs for adding dropout layers.""" HIDDEN_LAYERS = [{ 'units': 64 }, { 'activation': 'relu' }, { 'dropout': 0.1 }, { 'units': 64, 'activation': 'relu', 'name': 'relu2', 'dropout': 0.1 }] model = PhysicsGuidedNeuralNetwork(p_fun=p_fun_pythag, hidden_layers=HIDDEN_LAYERS, loss_weights=(0.0, 1.0), n_features=2, n_labels=1) assert len(model.layers) == 8, "dropout layers did not get added!" assert isinstance(model.layers[0], InputLayer) assert isinstance(model.layers[1], Dense) assert isinstance(model.layers[2], Activation) assert isinstance(model.layers[3], Dropout) assert isinstance(model.layers[4], Dense) assert isinstance(model.layers[5], Activation) assert isinstance(model.layers[6], Dropout) model.fit(X, Y_NOISE, P, n_batch=4, n_epoch=20) y_pred = model.predict(X) model.save(FPATH) loaded = PhysicsGuidedNeuralNetwork.load(FPATH) y_pred_loaded = loaded.predict(X) assert np.allclose(y_pred, y_pred_loaded) assert len(model.layers) == len(loaded.layers) os.remove(FPATH)
def test_dropouts(): """Test the dropout rate kwargs for adding dropout layers.""" hidden_layers_1 = [{ 'units': 64 }, { 'activation': 'relu' }, { 'units': 64, 'activation': 'relu', 'name': 'relu2' }] hidden_layers_2 = [{ 'units': 64 }, { 'activation': 'relu' }, { 'dropout': 0.1 }, { 'units': 64, 'activation': 'relu', 'name': 'relu2', 'dropout': 0.1 }] PhysicsGuidedNeuralNetwork.seed() model_1 = PhysicsGuidedNeuralNetwork(p_fun=p_fun_pythag, hidden_layers=hidden_layers_1, loss_weights=(0.0, 1.0), n_features=2, n_labels=1) PhysicsGuidedNeuralNetwork.seed() model_2 = PhysicsGuidedNeuralNetwork(p_fun=p_fun_pythag, hidden_layers=hidden_layers_2, loss_weights=(0.0, 1.0), n_features=2, n_labels=1) assert len(model_1.layers) == 6 assert len(model_2.layers) == 8, "dropout layers did not get added!" assert isinstance(model_2.layers[0], InputLayer) assert isinstance(model_2.layers[1], Dense) assert isinstance(model_2.layers[2], Activation) assert isinstance(model_2.layers[3], Dropout) assert isinstance(model_2.layers[4], Dense) assert isinstance(model_2.layers[5], Activation) assert isinstance(model_2.layers[6], Dropout) PhysicsGuidedNeuralNetwork.seed() model_1.fit(X, Y_NOISE, P, n_batch=4, n_epoch=20) PhysicsGuidedNeuralNetwork.seed() model_2.fit(X, Y_NOISE, P, n_batch=4, n_epoch=20) y_pred_1 = model_1.predict(X) y_pred_2 = model_2.predict(X) # make sure dropouts dont predict the same as non-dropout diff = np.abs(y_pred_1 - y_pred_2) assert not np.allclose(y_pred_1, y_pred_2) assert np.max(diff) > 0.1 with tempfile.TemporaryDirectory() as td: fpath = os.path.join(td, 'tempfile.pkl') model_2.save(fpath) loaded = PhysicsGuidedNeuralNetwork.load(fpath) y_pred_loaded = loaded.predict(X) assert np.allclose(y_pred_2, y_pred_loaded) assert len(model_2.layers) == len(loaded.layers)