def learn(epochs, verbose_epoch, X_train, y_train, test, learning_rate, weight_decay, batch_size):
    'Trains the model and predicts on the test data set.'
    net = get_net()
    _ = train(net, X_train, y_train, epochs, verbose_epoch, learning_rate, weight_decay, batch_size)
    preds = net(X_test).asnumpy()
    test['SalePrice'] = pd.Series(preds.reshape(1, (- 1))[0])
    submission = pd.concat([test['Id'], test['SalePrice']], axis=1)
    submission.to_csv('submission.csv', index=False)
Trains the model and predicts on the test data set.
def learn(epochs, verbose_epoch, X_train, y_train, test, learning_rate, weight_decay, batch_size):
    
    net = get_net()
    _ = train(net, X_train, y_train, epochs, verbose_epoch, learning_rate, weight_decay, batch_size)
    preds = net(X_test).asnumpy()
    test['SalePrice'] = pd.Series(preds.reshape(1, (- 1))[0])
    submission = pd.concat([test['Id'], test['SalePrice']], axis=1)
    submission.to_csv('submission.csv', index=False)