-
Book Overview & Buying
-
Table Of Contents
Machine Learning for Finance
By :
Our data now contains the following columns:
amount, oldBalanceOrig, newBalanceOrig, oldBalanceDest, newBalanceDest, isFraud, isFlaggedFraud, type_CASH_OUT, type_TRANSFER, isNight
Now that we've got the columns, our data is prepared, and we can use it to create a model.
To train the model, a neural network needs a target. In our case, isFraud is the target, so we have to separate it from the rest of the data. We can do this by running:
y_df = df['isFraud']
x_df = df.drop('isFraud',axis=1)The first step only returns the isFraud column and assigns it to y_df.
The second step returns all columns except isFraud and assigns them to x_df.
We also need to convert our data from a pandas DataFrame to NumPy arrays. The pandas DataFrame is built on top of NumPy arrays but comes with lots of extra bells and whistles that make all the preprocessing we did earlier possible. To train a neural network, however, we just need the underlying data...