Replace xgboost with PyTorch

[bacemkarray]

Now that the graph is functioning properly, it's time to begin increasing the complexity.

The current toy model from xgboost should be replaced with something more complex.

Not necessarily a priority right now, but something that should be tended to in the near future.