code: 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)
docs: Trains the model and predicts on the test data set.