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()