def predict(test):
"""Return prediction from test set"""
X, y = load_data.training()
net1.fit(X, y)
return net1.predict(test)
import numpy as np import pandas as pd import load_data features_train, target_train = load_data.training() features_test = load_data.test()
