Announcing the TensorFlow Dev Summit 2020 Learn more

Classification on imbalanced data

View on TensorFlow.org Run in Google Colab View source on GitHub Download notebook

This tutorial demonstrates how to classify a highly imbalanced dataset in which the number of examples in one class greatly outnumbers the examples in another. You will work with the Credit Card Fraud Detection dataset hosted on Kaggle. The aim is to detect a mere 492 fraudulent transactions from 284,807 transactions in total. You will use Keras to define the model and class weights to help the model learn from the imbalanced data. .

This tutorial contains complete code to:

  • Load a CSV file using Pandas.
  • Create train, validation, and test sets.
  • Define and train a model using Keras (including setting class weights).
  • Evaluate the model using various metrics (including precision and recall).
  • Try common tecniques for dealing with imbalanced data like:
    • Class weighting
    • Oversampling

Setup

from __future__ import absolute_import, division, print_function, unicode_literals

import tensorflow as tf
from tensorflow import keras

import os
import tempfile

import matplotlib as mpl
import matplotlib.pyplot as plt
import numpy as np
import pandas as pd
import seaborn as sns

import sklearn
from sklearn.metrics import confusion_matrix
from sklearn.model_selection import train_test_split
from sklearn.preprocessing import StandardScaler

mpl.rcParams['figure.figsize'] = (12, 10)
colors = plt.rcParams['axes.prop_cycle'].by_key()['color']

Data processing and exploration

Download the Kaggle Credit Card Fraud data set

Pandas is a Python library with many helpful utilities for loading and working with structured data and can be used to download CSVs into a dataframe.

file = tf.keras.utils
raw_df = pd.read_csv('https://storage.googleapis.com/download.tensorflow.org/data/creditcard.csv')
raw_df.head()

raw_df[['Time', 'V1', 'V2', 'V3', 'V4', 'V5', 'V26', 'V27', 'V28', 'Amount', 'Class']].describe()

Examine the class label imbalance

Let's look at the dataset imbalance:

neg, pos = np.bincount(raw_df['Class'])
total = neg + pos
print('Examples:\n    Total: {}\n    Positive: {} ({:.2f}% of total)\n'.format(
    total, pos, 100 * pos / total))

Examples:
    Total: 284807
    Positive: 492 (0.17% of total)

This shows the small fraction of positive samples.

Clean, split and normalize the data

The raw data has a few issues. First the Time and Amount columns are too variable to use directly. Drop the Time column (since it's not clear what it means) and take the log of the Amount column to reduce its range.

cleaned_df = raw_df.copy()

# You don't want the `Time` column.
cleaned_df.pop('Time')

# The `Amount` column covers a huge range. Convert to log-space.
eps=0.001 # 0 => 0.1¢
cleaned_df['Log Ammount'] = np.log(cleaned_df.pop('Amount')+eps)

Split the dataset into train, validation, and test sets. The validation set is used during the model fitting to evaluate the loss and any metrics, however the model is not fit with this data. The test set is completely unused during the training phase and is only used at the end to evaluate how well the model generalizes to new data. This is especially important with imbalanced datasets where overfitting is a significant concern from the lack of training data.

# Use a utility from sklearn to split and shuffle our dataset.
train_df, test_df = train_test_split(cleaned_df, test_size=0.2)
train_df, val_df = train_test_split(train_df, test_size=0.2)

# Form np arrays of labels and features.
train_labels = np.array(train_df.pop('Class'))
bool_train_labels = train_labels != 0
val_labels = np.array(val_df.pop('Class'))
test_labels = np.array(test_df.pop('Class'))

train_features = np.array(train_df)
val_features = np.array(val_df)
test_features = np.array(test_df)

Normalize the input features using the sklearn StandardScaler. This will set the mean to 0 and standard deviation to 1.

scaler = StandardScaler()
train_features = scaler.fit_transform(train_features)

val_features = scaler.transform(val_features)
test_features = scaler.transform(test_features)

train_features = np.clip(train_features, -5, 5)
val_features = np.clip(val_features, -5, 5)
test_features = np.clip(test_features, -5, 5)


print('Training labels shape:', train_labels.shape)
print('Validation labels shape:', val_labels.shape)
print('Test labels shape:', test_labels.shape)

print('Training features shape:', train_features.shape)
print('Validation features shape:', val_features.shape)
print('Test features shape:', test_features.shape)


Training labels shape: (182276,)
Validation labels shape: (45569,)
Test labels shape: (56962,)
Training features shape: (182276, 29)
Validation features shape: (45569, 29)
Test features shape: (56962, 29)

Look at the data distribution

Next compare the distributions of the positive and negative examples over a few features. Good questions to ask yourself at this point are:

  • Do these distributions make sense?
    • Yes. You've normalized the input and these are mostly concentrated in the +/- 2 range.
  • Can you see the difference between the ditributions?
    • Yes the positive examples contain a much higher rate of extreme values.
pos_df = pd.DataFrame(train_features[ bool_train_labels], columns = train_df.columns)
neg_df = pd.DataFrame(train_features[~bool_train_labels], columns = train_df.columns)

sns.jointplot(pos_df['V5'], pos_df['V6'],
              kind='hex', xlim = (-5,5), ylim = (-5,5))
plt.suptitle("Positive distribution")

sns.jointplot(neg_df['V5'], neg_df['V6'],
              kind='hex', xlim = (-5,5), ylim = (-5,5))
_ = plt.suptitle("Negative distribution")

png

png

Define the model and metrics

Define a function that creates a simple neural network with a densly connected hidden layer, a dropout layer to reduce overfitting, and an output sigmoid layer that returns the probability of a transaction being fraudulent: 

METRICS = [
      keras.metrics.TruePositives(name='tp'),
      keras.metrics.FalsePositives(name='fp'),
      keras.metrics.TrueNegatives(name='tn'),
      keras.metrics.FalseNegatives(name='fn'), 
      keras.metrics.BinaryAccuracy(name='accuracy'),
      keras.metrics.Precision(name='precision'),
      keras.metrics.Recall(name='recall'),
      keras.metrics.AUC(name='auc'),
]

def make_model(metrics = METRICS, output_bias=None):
  if output_bias is not None:
    output_bias = tf.keras.initializers.Constant(output_bias)
  model = keras.Sequential([
      keras.layers.Dense(
          16, activation='relu',
          input_shape=(train_features.shape[-1],)),
      keras.layers.Dropout(0.5),
      keras.layers.Dense(1, activation='sigmoid',
                         bias_initializer=output_bias),
  ])

  model.compile(
      optimizer=keras.optimizers.Adam(lr=1e-3),
      loss=keras.losses.BinaryCrossentropy(),
      metrics=metrics)

  return model

Understanding useful metrics

Notice that there are a few metrics defined above that can be computed by the model that will be helpful when evaluating the performance.

  • False negatives and false positives are samples that were incorrectly classified
  • True negatives and true positives are samples that were correctly classified
  • Accuracy is the percentage of examples correctly classified > $\frac{\text{true samples}}{\text{total samples}}$
  • Precision is the percentage of predicted positives that were correctly classified > $\frac{\text{true positives}}{\text{true positives + false positives}}$
  • Recall is the percentage of actual positives that were correctly classified > $\frac{\text{true positives}}{\text{true positives + false negatives}}$
  • AUC refers to the Area Under the Curve of a Receiver Operating Characteristic curve (ROC-AUC). This metric is equal to the probability that a classifier will rank a random positive sample higher than than a random negative sample.

Read more:

Baseline model

Build the model

Now create and train your model using the function that was defined earlier. Notice that the model is fit using a larger than default batch size of 2048, this is important to ensure that each batch has a decent chance of containing a few positive samples. If the batch size was too small, they would likely have no fraudulent transactions to learn from.

EPOCHS = 100
BATCH_SIZE = 2048

early_stopping = tf.keras.callbacks.EarlyStopping(
    monitor='val_auc', 
    verbose=1,
    patience=10,
    mode='max',
    restore_best_weights=True)

model = make_model()
model.summary()

Model: "sequential"
_________________________________________________________________
Layer (type)                 Output Shape              Param #   
=================================================================
dense (Dense)                (None, 16)                480       
_________________________________________________________________
dropout (Dropout)            (None, 16)                0         
_________________________________________________________________
dense_1 (Dense)              (None, 1)                 17        
=================================================================
Total params: 497
Trainable params: 497
Non-trainable params: 0
_________________________________________________________________

Test run the model:

model.predict(train_features[:10])

array([[0.6296253 ],
       [0.82509124],
       [0.75135857],
       [0.73724824],
       [0.82174015],
       [0.33519754],
       [0.6719973 ],
       [0.30910844],
       [0.6378555 ],
       [0.8381703 ]], dtype=float32)

Optional: Set the correct initial bias.

These are initial guesses are not great. You know the dataset is imbalanced. Set the output layer's bias to reflect that (See: A Recipe for Training Neural Networks: "init well"). This can help with initial convergence.

With the default bias initialization the loss should be about math.log(2) = 0.69314 

results = model.evaluate(train_features, train_labels, batch_size=BATCH_SIZE, verbose=0)
print("Loss: {:0.4f}".format(results[0]))

Loss: 1.0921

The correct bias to set can be derived from:

$$ p_0 = pos/(pos + neg) = 1/(1+e^{-b_0}) $$
$$ b_0 = -log_e(1/p_0 - 1) $$
$$ b_0 = log_e(pos/neg)$$ 
initial_bias = np.log([pos/neg])
initial_bias

array([-6.35935934])

Set that as the initial bias, and the model will give much more reasonable initial guesses.

It should be near: pos/total = 0.0018

model = make_model(output_bias = initial_bias)
model.predict(train_features[:10])

array([[0.00124893],
       [0.00185736],
       [0.00164955],
       [0.00123761],
       [0.00137692],
       [0.00182851],
       [0.00170887],
       [0.00239349],
       [0.0024704 ],
       [0.00517672]], dtype=float32)

With this initialization the initial loss should be approximately:

$$-p_0log(p_0)-(1-p_0)log(1-p_0) = 0.01317$$ 
results = model.evaluate(train_features, train_labels, batch_size=BATCH_SIZE, verbose=0)
print("Loss: {:0.4f}".format(results[0]))

Loss: 0.0157

This initial loss is about 50 times less than if would have been with naive initilization.

This way the model doesn't need to spend the first few epochs just learning that positive examples are unlikely. This also makes it easier to read plots of the loss during training.

Checkpoint the initial weights

To make the various training runs more comparable, keep this initial model's weights in a checkpoint file, and load them into each model before training.

initial_weights = os.path.join(tempfile.mkdtemp(),'initial_weights')
model.save_weights(initial_weights)

Confirm that the bias fix helps

Before moving on, confirm quick that the careful bias initialization actually helped.

Train the model for 20 epochs, with and without this careful initialization, and compare the losses: 

model = make_model()
model.load_weights(initial_weights)
model.layers[-1].bias.assign([0.0])
zero_bias_history = model.fit(
    train_features,
    train_labels,
    batch_size=BATCH_SIZE,
    epochs=20,
    validation_data=(val_features, val_labels), 
    verbose=0)

model = make_model()
model.load_weights(initial_weights)
careful_bias_history = model.fit(
    train_features,
    train_labels,
    batch_size=BATCH_SIZE,
    epochs=20,
    validation_data=(val_features, val_labels), 
    verbose=0)

def plot_loss(history, label, n):
  # Use a log scale to show the wide range of values.
  plt.semilogy(history.epoch,  history.history['loss'],
               color=colors[n], label='Train '+label)
  plt.semilogy(history.epoch,  history.history['val_loss'],
          color=colors[n], label='Val '+label,
          linestyle="--")
  plt.xlabel('Epoch')
  plt.ylabel('Loss')
  
  plt.legend()

plot_loss(zero_bias_history, "Zero Bias", 0)
plot_loss(careful_bias_history, "Careful Bias", 1)

png

The above figure makes it clear: In terms of validation loss, on this problem, this careful initialization gives a clear advantage.

Train the model

model = make_model()
model.load_weights(initial_weights)
baseline_history = model.fit(
    train_features,
    train_labels,
    batch_size=BATCH_SIZE,
    epochs=EPOCHS,
    callbacks = [early_stopping],
    validation_data=(val_features, val_labels))

Train on 182276 samples, validate on 45569 samples
Epoch 1/100
182276/182276 [==============================] - 2s 12us/sample - loss: 0.0139 - tp: 7.0000 - fp: 124.0000 - tn: 181835.0000 - fn: 310.0000 - accuracy: 0.9976 - precision: 0.0534 - recall: 0.0221 - auc: 0.7262 - val_loss: 0.0074 - val_tp: 4.0000 - val_fp: 0.0000e+00 - val_tn: 45492.0000 - val_fn: 73.0000 - val_accuracy: 0.9984 - val_precision: 1.0000 - val_recall: 0.0519 - val_auc: 0.8742
Epoch 2/100
182276/182276 [==============================] - 0s 3us/sample - loss: 0.0076 - tp: 91.0000 - fp: 30.0000 - tn: 181929.0000 - fn: 226.0000 - accuracy: 0.9986 - precision: 0.7521 - recall: 0.2871 - auc: 0.8828 - val_loss: 0.0053 - val_tp: 39.0000 - val_fp: 7.0000 - val_tn: 45485.0000 - val_fn: 38.0000 - val_accuracy: 0.9990 - val_precision: 0.8478 - val_recall: 0.5065 - val_auc: 0.8761
Epoch 3/100
182276/182276 [==============================] - 0s 3us/sample - loss: 0.0064 - tp: 146.0000 - fp: 36.0000 - tn: 181923.0000 - fn: 171.0000 - accuracy: 0.9989 - precision: 0.8022 - recall: 0.4606 - auc: 0.8981 - val_loss: 0.0049 - val_tp: 45.0000 - val_fp: 7.0000 - val_tn: 45485.0000 - val_fn: 32.0000 - val_accuracy: 0.9991 - val_precision: 0.8654 - val_recall: 0.5844 - val_auc: 0.8828
Epoch 4/100
182276/182276 [==============================] - 0s 3us/sample - loss: 0.0057 - tp: 167.0000 - fp: 33.0000 - tn: 181926.0000 - fn: 150.0000 - accuracy: 0.9990 - precision: 0.8350 - recall: 0.5268 - auc: 0.9183 - val_loss: 0.0046 - val_tp: 47.0000 - val_fp: 8.0000 - val_tn: 45484.0000 - val_fn: 30.0000 - val_accuracy: 0.9992 - val_precision: 0.8545 - val_recall: 0.6104 - val_auc: 0.8829
Epoch 5/100
182276/182276 [==============================] - 0s 3us/sample - loss: 0.0051 - tp: 184.0000 - fp: 37.0000 - tn: 181922.0000 - fn: 133.0000 - accuracy: 0.9991 - precision: 0.8326 - recall: 0.5804 - auc: 0.9159 - val_loss: 0.0045 - val_tp: 51.0000 - val_fp: 8.0000 - val_tn: 45484.0000 - val_fn: 26.0000 - val_accuracy: 0.9993 - val_precision: 0.8644 - val_recall: 0.6623 - val_auc: 0.8893
Epoch 6/100
182276/182276 [==============================] - 0s 3us/sample - loss: 0.0052 - tp: 163.0000 - fp: 37.0000 - tn: 181922.0000 - fn: 154.0000 - accuracy: 0.9990 - precision: 0.8150 - recall: 0.5142 - auc: 0.9242 - val_loss: 0.0044 - val_tp: 51.0000 - val_fp: 8.0000 - val_tn: 45484.0000 - val_fn: 26.0000 - val_accuracy: 0.9993 - val_precision: 0.8644 - val_recall: 0.6623 - val_auc: 0.8957
Epoch 7/100
182276/182276 [==============================] - 0s 3us/sample - loss: 0.0049 - tp: 186.0000 - fp: 30.0000 - tn: 181929.0000 - fn: 131.0000 - accuracy: 0.9991 - precision: 0.8611 - recall: 0.5868 - auc: 0.9211 - val_loss: 0.0042 - val_tp: 52.0000 - val_fp: 8.0000 - val_tn: 45484.0000 - val_fn: 25.0000 - val_accuracy: 0.9993 - val_precision: 0.8667 - val_recall: 0.6753 - val_auc: 0.8957
Epoch 8/100
182276/182276 [==============================] - 0s 3us/sample - loss: 0.0049 - tp: 170.0000 - fp: 36.0000 - tn: 181923.0000 - fn: 147.0000 - accuracy: 0.9990 - precision: 0.8252 - recall: 0.5363 - auc: 0.9277 - val_loss: 0.0041 - val_tp: 53.0000 - val_fp: 8.0000 - val_tn: 45484.0000 - val_fn: 24.0000 - val_accuracy: 0.9993 - val_precision: 0.8689 - val_recall: 0.6883 - val_auc: 0.8957
Epoch 9/100
182276/182276 [==============================] - 0s 3us/sample - loss: 0.0045 - tp: 199.0000 - fp: 29.0000 - tn: 181930.0000 - fn: 118.0000 - accuracy: 0.9992 - precision: 0.8728 - recall: 0.6278 - auc: 0.9325 - val_loss: 0.0041 - val_tp: 53.0000 - val_fp: 8.0000 - val_tn: 45484.0000 - val_fn: 24.0000 - val_accuracy: 0.9993 - val_precision: 0.8689 - val_recall: 0.6883 - val_auc: 0.8957
Epoch 10/100
182276/182276 [==============================] - 0s 3us/sample - loss: 0.0044 - tp: 190.0000 - fp: 31.0000 - tn: 181928.0000 - fn: 127.0000 - accuracy: 0.9991 - precision: 0.8597 - recall: 0.5994 - auc: 0.9295 - val_loss: 0.0041 - val_tp: 54.0000 - val_fp: 8.0000 - val_tn: 45484.0000 - val_fn: 23.0000 - val_accuracy: 0.9993 - val_precision: 0.8710 - val_recall: 0.7013 - val_auc: 0.8957
Epoch 11/100
182276/182276 [==============================] - 0s 3us/sample - loss: 0.0044 - tp: 191.0000 - fp: 38.0000 - tn: 181921.0000 - fn: 126.0000 - accuracy: 0.9991 - precision: 0.8341 - recall: 0.6025 - auc: 0.9279 - val_loss: 0.0040 - val_tp: 53.0000 - val_fp: 8.0000 - val_tn: 45484.0000 - val_fn: 24.0000 - val_accuracy: 0.9993 - val_precision: 0.8689 - val_recall: 0.6883 - val_auc: 0.8957
Epoch 12/100
182276/182276 [==============================] - 0s 3us/sample - loss: 0.0045 - tp: 178.0000 - fp: 31.0000 - tn: 181928.0000 - fn: 139.0000 - accuracy: 0.9991 - precision: 0.8517 - recall: 0.5615 - auc: 0.9217 - val_loss: 0.0040 - val_tp: 54.0000 - val_fp: 8.0000 - val_tn: 45484.0000 - val_fn: 23.0000 - val_accuracy: 0.9993 - val_precision: 0.8710 - val_recall: 0.7013 - val_auc: 0.8957
Epoch 13/100
182276/182276 [==============================] - 0s 3us/sample - loss: 0.0044 - tp: 181.0000 - fp: 39.0000 - tn: 181920.0000 - fn: 136.0000 - accuracy: 0.9990 - precision: 0.8227 - recall: 0.5710 - auc: 0.9310 - val_loss: 0.0040 - val_tp: 54.0000 - val_fp: 8.0000 - val_tn: 45484.0000 - val_fn: 23.0000 - val_accuracy: 0.9993 - val_precision: 0.8710 - val_recall: 0.7013 - val_auc: 0.8957
Epoch 14/100
182276/182276 [==============================] - 0s 3us/sample - loss: 0.0042 - tp: 187.0000 - fp: 37.0000 - tn: 181922.0000 - fn: 130.0000 - accuracy: 0.9991 - precision: 0.8348 - recall: 0.5899 - auc: 0.9296 - val_loss: 0.0039 - val_tp: 54.0000 - val_fp: 8.0000 - val_tn: 45484.0000 - val_fn: 23.0000 - val_accuracy: 0.9993 - val_precision: 0.8710 - val_recall: 0.7013 - val_auc: 0.8957
Epoch 15/100
182276/182276 [==============================] - 0s 3us/sample - loss: 0.0041 - tp: 186.0000 - fp: 27.0000 - tn: 181932.0000 - fn: 131.0000 - accuracy: 0.9991 - precision: 0.8732 - recall: 0.5868 - auc: 0.9359 - val_loss: 0.0039 - val_tp: 56.0000 - val_fp: 8.0000 - val_tn: 45484.0000 - val_fn: 21.0000 - val_accuracy: 0.9994 - val_precision: 0.8750 - val_recall: 0.7273 - val_auc: 0.8957
Epoch 16/100
159744/182276 [=========================>....] - ETA: 0s - loss: 0.0041 - tp: 165.0000 - fp: 31.0000 - tn: 159440.0000 - fn: 108.0000 - accuracy: 0.9991 - precision: 0.8418 - recall: 0.6044 - auc: 0.9257Restoring model weights from the end of the best epoch.
182276/182276 [==============================] - 0s 3us/sample - loss: 0.0042 - tp: 192.0000 - fp: 35.0000 - tn: 181924.0000 - fn: 125.0000 - accuracy: 0.9991 - precision: 0.8458 - recall: 0.6057 - auc: 0.9265 - val_loss: 0.0039 - val_tp: 54.0000 - val_fp: 8.0000 - val_tn: 45484.0000 - val_fn: 23.0000 - val_accuracy: 0.9993 - val_precision: 0.8710 - val_recall: 0.7013 - val_auc: 0.8957
Epoch 00016: early stopping

Check training history

In this section, you will produce plots of your model's accuracy and loss on the training and validation set. These are useful to check for overfitting, which you can learn more about in this tutorial.

Additionally, you can produce these plots for any of the metrics you created above. False negatives are included as an example.

def plot_metrics(history):
  metrics =  ['loss', 'auc', 'precision', 'recall']
  for n, metric in enumerate(metrics):
    name = metric.replace("_"," ").capitalize()
    plt.subplot(2,2,n+1)
    plt.plot(history.epoch,  history.history[metric], color=colors[0], label='Train')
    plt.plot(history.epoch, history.history['val_'+metric],
             color=colors[0], linestyle="--", label='Val')
    plt.xlabel('Epoch')
    plt.ylabel(name)
    if metric == 'loss':
      plt.ylim([0, plt.ylim()[1]])
    elif metric == 'auc':
      plt.ylim([0.8,1])
    else:
      plt.ylim([0,1])

    plt.legend()


plot_metrics(baseline_history)

png

Evaluate metrics

You can use a confusion matrix to summarize the actual vs. predicted labels where the X axis is the predicted label and the Y axis is the actual label.

train_predictions_baseline = model.predict(train_features, batch_size=BATCH_SIZE)
test_predictions_baseline = model.predict(test_features, batch_size=BATCH_SIZE)

def plot_cm(labels, predictions, p=0.5):
  cm = confusion_matrix(labels, predictions > p)
  plt.figure(figsize=(5,5))
  sns.heatmap(cm, annot=True, fmt="d")
  plt.title('Confusion matrix @{:.2f}'.format(p))
  plt.ylabel('Actual label')
  plt.xlabel('Predicted label')

  print('Legitimate Transactions Detected (True Negatives): ', cm[0][0])
  print('Legitimate Transactions Incorrectly Detected (False Positives): ', cm[0][1])
  print('Fraudulent Transactions Missed (False Negatives): ', cm[1][0])
  print('Fraudulent Transactions Detected (True Positives): ', cm[1][1])
  print('Total Fraudulent Transactions: ', np.sum(cm[1]))

Evaluate your model on the test dataset and display the results for the metrics you created above.

baseline_results = model.evaluate(test_features, test_labels,
                                  batch_size=BATCH_SIZE, verbose=0)
for name, value in zip(model.metrics_names, baseline_results):
  print(name, ': ', value)
print()

plot_cm(test_labels, test_predictions_baseline)

loss :  0.003458617485003855
tp :  70.0
fp :  9.0
tn :  56855.0
fn :  28.0
accuracy :  0.9993504
precision :  0.886076
recall :  0.71428573
auc :  0.9436206

Legitimate Transactions Detected (True Negatives):  56855
Legitimate Transactions Incorrectly Detected (False Positives):  9
Fraudulent Transactions Missed (False Negatives):  28
Fraudulent Transactions Detected (True Positives):  70
Total Fraudulent Transactions:  98

png

If the model had predicted everything perfectly, this would be a diagonal matrix where values off the main diagonal, indicating incorrect predictions, would be zero. In this case the matrix shows that you have relatively few false positives, meaning that there were relatively few legitimate transactions that were incorrectly flagged. However, you would likely want to have even fewer false negatives despite the cost of increasing the number of false positives. This trade off may be preferable because false negatives would allow fraudulent transactions to go through, whereas false positives may cause an email to be sent to a customer to ask them to verify their card activity.

Plot the ROC

Now plot the ROC. This plot is useful because it shows, at a glance, the range of performance the model can reach just by tuning the output threshold.

def plot_roc(name, labels, predictions, **kwargs):
  fp, tp, _ = sklearn.metrics.roc_curve(labels, predictions)

  plt.plot(100*fp, 100*tp, label=name, linewidth=2, **kwargs)
  plt.xlabel('False positives [%]')
  plt.ylabel('True positives [%]')
  plt.xlim([-0.5,20])
  plt.ylim([80,100.5])
  plt.grid(True)
  ax = plt.gca()
  ax.set_aspect('equal')

plot_roc("Train Baseline", train_labels, train_predictions_baseline, color=colors[0])
plot_roc("Test Baseline", test_labels, test_predictions_baseline, color=colors[0], linestyle='--')
plt.legend(loc='lower right')

<matplotlib.legend.Legend at 0x7ff3052739e8>

png

It looks like the precision is relatively high, but the recall and the area under the ROC curve (AUC) aren't as high as you might like. Classifiers often face challenges when trying to maximize both precision and recall, which is especially true when working with imbalanced datasets. It is important to consider the costs of different types of errors in the context of the problem you care about. In this example, a false negative (a fraudulent transaction is missed) may have a financial cost, while a false positive (a transaction is incorrectly flagged as fraudulent) may decrease user happiness.

Class weights

Calculate class weights

The goal is to identify fradulent transactions, but you don't have very many of those positive samples to work with, so you would want to have the classifier heavily weight the few examples that are available. You can do this by passing Keras weights for each class through a parameter. These will cause the model to "pay more attention" to examples from an under-represented class.

# Scaling by total/2 helps keep the loss to a similar magnitude.
# The sum of the weights of all examples stays the same.
weight_for_0 = (1 / neg)*(total)/2.0 
weight_for_1 = (1 / pos)*(total)/2.0

class_weight = {0: weight_for_0, 1: weight_for_1}

print('Weight for class 0: {:.2f}'.format(weight_for_0))
print('Weight for class 1: {:.2f}'.format(weight_for_1))

Weight for class 0: 0.50
Weight for class 1: 289.44

Train a model with class weights

Now try re-training and evaluating the model with class weights to see how that affects the predictions.

weighted_model = make_model()
weighted_model.load_weights(initial_weights)

weighted_history = weighted_model.fit(
    train_features,
    train_labels,
    batch_size=BATCH_SIZE,
    epochs=EPOCHS,
    callbacks = [early_stopping],
    validation_data=(val_features, val_labels),
    # The class weights go here
    class_weight=class_weight) 

WARNING:tensorflow:sample_weight modes were coerced from
  ...
    to  
  ['...']
WARNING:tensorflow:sample_weight modes were coerced from
  ...
    to  
  ['...']
Train on 182276 samples, validate on 45569 samples
Epoch 1/100
182276/182276 [==============================] - 2s 13us/sample - loss: 1.9444 - tp: 43.0000 - fp: 678.0000 - tn: 181281.0000 - fn: 274.0000 - accuracy: 0.9948 - precision: 0.0596 - recall: 0.1356 - auc: 0.7373 - val_loss: 0.8463 - val_tp: 38.0000 - val_fp: 76.0000 - val_tn: 45416.0000 - val_fn: 39.0000 - val_accuracy: 0.9975 - val_precision: 0.3333 - val_recall: 0.4935 - val_auc: 0.9035
Epoch 2/100
182276/182276 [==============================] - 0s 3us/sample - loss: 0.8102 - tp: 178.0000 - fp: 1298.0000 - tn: 180661.0000 - fn: 139.0000 - accuracy: 0.9921 - precision: 0.1206 - recall: 0.5615 - auc: 0.9017 - val_loss: 0.5681 - val_tp: 58.0000 - val_fp: 151.0000 - val_tn: 45341.0000 - val_fn: 19.0000 - val_accuracy: 0.9963 - val_precision: 0.2775 - val_recall: 0.7532 - val_auc: 0.9279
Epoch 3/100
182276/182276 [==============================] - 0s 3us/sample - loss: 0.5391 - tp: 233.0000 - fp: 1916.0000 - tn: 180043.0000 - fn: 84.0000 - accuracy: 0.9890 - precision: 0.1084 - recall: 0.7350 - auc: 0.9164 - val_loss: 0.4939 - val_tp: 58.0000 - val_fp: 228.0000 - val_tn: 45264.0000 - val_fn: 19.0000 - val_accuracy: 0.9946 - val_precision: 0.2028 - val_recall: 0.7532 - val_auc: 0.9423
Epoch 4/100
182276/182276 [==============================] - 0s 3us/sample - loss: 0.4399 - tp: 244.0000 - fp: 2584.0000 - tn: 179375.0000 - fn: 73.0000 - accuracy: 0.9854 - precision: 0.0863 - recall: 0.7697 - auc: 0.9346 - val_loss: 0.4412 - val_tp: 58.0000 - val_fp: 316.0000 - val_tn: 45176.0000 - val_fn: 19.0000 - val_accuracy: 0.9926 - val_precision: 0.1551 - val_recall: 0.7532 - val_auc: 0.9456
Epoch 5/100
182276/182276 [==============================] - 0s 3us/sample - loss: 0.3779 - tp: 256.0000 - fp: 3158.0000 - tn: 178801.0000 - fn: 61.0000 - accuracy: 0.9823 - precision: 0.0750 - recall: 0.8076 - auc: 0.9369 - val_loss: 0.4008 - val_tp: 58.0000 - val_fp: 382.0000 - val_tn: 45110.0000 - val_fn: 19.0000 - val_accuracy: 0.9912 - val_precision: 0.1318 - val_recall: 0.7532 - val_auc: 0.9525
Epoch 6/100
182276/182276 [==============================] - 0s 3us/sample - loss: 0.3094 - tp: 267.0000 - fp: 3839.0000 - tn: 178120.0000 - fn: 50.0000 - accuracy: 0.9787 - precision: 0.0650 - recall: 0.8423 - auc: 0.9559 - val_loss: 0.3680 - val_tp: 60.0000 - val_fp: 496.0000 - val_tn: 44996.0000 - val_fn: 17.0000 - val_accuracy: 0.9887 - val_precision: 0.1079 - val_recall: 0.7792 - val_auc: 0.9536
Epoch 7/100
182276/182276 [==============================] - 0s 3us/sample - loss: 0.3012 - tp: 267.0000 - fp: 4441.0000 - tn: 177518.0000 - fn: 50.0000 - accuracy: 0.9754 - precision: 0.0567 - recall: 0.8423 - auc: 0.9579 - val_loss: 0.3392 - val_tp: 60.0000 - val_fp: 717.0000 - val_tn: 44775.0000 - val_fn: 17.0000 - val_accuracy: 0.9839 - val_precision: 0.0772 - val_recall: 0.7792 - val_auc: 0.9569
Epoch 8/100
182276/182276 [==============================] - 0s 3us/sample - loss: 0.2576 - tp: 273.0000 - fp: 5089.0000 - tn: 176870.0000 - fn: 44.0000 - accuracy: 0.9718 - precision: 0.0509 - recall: 0.8612 - auc: 0.9626 - val_loss: 0.3179 - val_tp: 61.0000 - val_fp: 765.0000 - val_tn: 44727.0000 - val_fn: 16.0000 - val_accuracy: 0.9829 - val_precision: 0.0738 - val_recall: 0.7922 - val_auc: 0.9638
Epoch 9/100
182276/182276 [==============================] - 0s 3us/sample - loss: 0.2565 - tp: 281.0000 - fp: 5478.0000 - tn: 176481.0000 - fn: 36.0000 - accuracy: 0.9697 - precision: 0.0488 - recall: 0.8864 - auc: 0.9577 - val_loss: 0.3126 - val_tp: 61.0000 - val_fp: 752.0000 - val_tn: 44740.0000 - val_fn: 16.0000 - val_accuracy: 0.9831 - val_precision: 0.0750 - val_recall: 0.7922 - val_auc: 0.9627
Epoch 10/100
182276/182276 [==============================] - 0s 3us/sample - loss: 0.2696 - tp: 276.0000 - fp: 5699.0000 - tn: 176260.0000 - fn: 41.0000 - accuracy: 0.9685 - precision: 0.0462 - recall: 0.8707 - auc: 0.9577 - val_loss: 0.2997 - val_tp: 64.0000 - val_fp: 782.0000 - val_tn: 44710.0000 - val_fn: 13.0000 - val_accuracy: 0.9826 - val_precision: 0.0757 - val_recall: 0.8312 - val_auc: 0.9657
Epoch 11/100
182276/182276 [==============================] - 0s 3us/sample - loss: 0.2517 - tp: 279.0000 - fp: 5870.0000 - tn: 176089.0000 - fn: 38.0000 - accuracy: 0.9676 - precision: 0.0454 - recall: 0.8801 - auc: 0.9614 - val_loss: 0.2903 - val_tp: 64.0000 - val_fp: 800.0000 - val_tn: 44692.0000 - val_fn: 13.0000 - val_accuracy: 0.9822 - val_precision: 0.0741 - val_recall: 0.8312 - val_auc: 0.9663
Epoch 12/100
182276/182276 [==============================] - 0s 3us/sample - loss: 0.2228 - tp: 282.0000 - fp: 6162.0000 - tn: 175797.0000 - fn: 35.0000 - accuracy: 0.9660 - precision: 0.0438 - recall: 0.8896 - auc: 0.9711 - val_loss: 0.2844 - val_tp: 64.0000 - val_fp: 874.0000 - val_tn: 44618.0000 - val_fn: 13.0000 - val_accuracy: 0.9805 - val_precision: 0.0682 - val_recall: 0.8312 - val_auc: 0.9676
Epoch 13/100
182276/182276 [==============================] - 0s 3us/sample - loss: 0.2573 - tp: 280.0000 - fp: 6257.0000 - tn: 175702.0000 - fn: 37.0000 - accuracy: 0.9655 - precision: 0.0428 - recall: 0.8833 - auc: 0.9613 - val_loss: 0.2777 - val_tp: 64.0000 - val_fp: 876.0000 - val_tn: 44616.0000 - val_fn: 13.0000 - val_accuracy: 0.9805 - val_precision: 0.0681 - val_recall: 0.8312 - val_auc: 0.9684
Epoch 14/100
182276/182276 [==============================] - 0s 3us/sample - loss: 0.2208 - tp: 285.0000 - fp: 6434.0000 - tn: 175525.0000 - fn: 32.0000 - accuracy: 0.9645 - precision: 0.0424 - recall: 0.8991 - auc: 0.9694 - val_loss: 0.2722 - val_tp: 64.0000 - val_fp: 914.0000 - val_tn: 44578.0000 - val_fn: 13.0000 - val_accuracy: 0.9797 - val_precision: 0.0654 - val_recall: 0.8312 - val_auc: 0.9686
Epoch 15/100
182276/182276 [==============================] - 0s 3us/sample - loss: 0.2480 - tp: 279.0000 - fp: 6440.0000 - tn: 175519.0000 - fn: 38.0000 - accuracy: 0.9645 - precision: 0.0415 - recall: 0.8801 - auc: 0.9621 - val_loss: 0.2674 - val_tp: 64.0000 - val_fp: 929.0000 - val_tn: 44563.0000 - val_fn: 13.0000 - val_accuracy: 0.9793 - val_precision: 0.0645 - val_recall: 0.8312 - val_auc: 0.9687
Epoch 16/100
182276/182276 [==============================] - 0s 3us/sample - loss: 0.2176 - tp: 283.0000 - fp: 6242.0000 - tn: 175717.0000 - fn: 34.0000 - accuracy: 0.9656 - precision: 0.0434 - recall: 0.8927 - auc: 0.9698 - val_loss: 0.2664 - val_tp: 64.0000 - val_fp: 942.0000 - val_tn: 44550.0000 - val_fn: 13.0000 - val_accuracy: 0.9790 - val_precision: 0.0636 - val_recall: 0.8312 - val_auc: 0.9691
Epoch 17/100
182276/182276 [==============================] - 0s 3us/sample - loss: 0.2065 - tp: 287.0000 - fp: 6315.0000 - tn: 175644.0000 - fn: 30.0000 - accuracy: 0.9652 - precision: 0.0435 - recall: 0.9054 - auc: 0.9714 - val_loss: 0.2657 - val_tp: 64.0000 - val_fp: 913.0000 - val_tn: 44579.0000 - val_fn: 13.0000 - val_accuracy: 0.9797 - val_precision: 0.0655 - val_recall: 0.8312 - val_auc: 0.9720
Epoch 18/100
182276/182276 [==============================] - 0s 3us/sample - loss: 0.1979 - tp: 286.0000 - fp: 5962.0000 - tn: 175997.0000 - fn: 31.0000 - accuracy: 0.9671 - precision: 0.0458 - recall: 0.9022 - auc: 0.9734 - val_loss: 0.2655 - val_tp: 64.0000 - val_fp: 873.0000 - val_tn: 44619.0000 - val_fn: 13.0000 - val_accuracy: 0.9806 - val_precision: 0.0683 - val_recall: 0.8312 - val_auc: 0.9722
Epoch 19/100
182276/182276 [==============================] - 0s 3us/sample - loss: 0.2293 - tp: 281.0000 - fp: 6071.0000 - tn: 175888.0000 - fn: 36.0000 - accuracy: 0.9665 - precision: 0.0442 - recall: 0.8864 - auc: 0.9700 - val_loss: 0.2609 - val_tp: 64.0000 - val_fp: 875.0000 - val_tn: 44617.0000 - val_fn: 13.0000 - val_accuracy: 0.9805 - val_precision: 0.0682 - val_recall: 0.8312 - val_auc: 0.9716
Epoch 20/100
182276/182276 [==============================] - 1s 3us/sample - loss: 0.1968 - tp: 286.0000 - fp: 6095.0000 - tn: 175864.0000 - fn: 31.0000 - accuracy: 0.9664 - precision: 0.0448 - recall: 0.9022 - auc: 0.9730 - val_loss: 0.2595 - val_tp: 65.0000 - val_fp: 872.0000 - val_tn: 44620.0000 - val_fn: 12.0000 - val_accuracy: 0.9806 - val_precision: 0.0694 - val_recall: 0.8442 - val_auc: 0.9731
Epoch 21/100
182276/182276 [==============================] - 0s 3us/sample - loss: 0.2166 - tp: 285.0000 - fp: 6028.0000 - tn: 175931.0000 - fn: 32.0000 - accuracy: 0.9668 - precision: 0.0451 - recall: 0.8991 - auc: 0.9714 - val_loss: 0.2587 - val_tp: 65.0000 - val_fp: 879.0000 - val_tn: 44613.0000 - val_fn: 12.0000 - val_accuracy: 0.9804 - val_precision: 0.0689 - val_recall: 0.8442 - val_auc: 0.9736
Epoch 22/100
182276/182276 [==============================] - 0s 3us/sample - loss: 0.2130 - tp: 287.0000 - fp: 5943.0000 - tn: 176016.0000 - fn: 30.0000 - accuracy: 0.9672 - precision: 0.0461 - recall: 0.9054 - auc: 0.9684 - val_loss: 0.2557 - val_tp: 65.0000 - val_fp: 872.0000 - val_tn: 44620.0000 - val_fn: 12.0000 - val_accuracy: 0.9806 - val_precision: 0.0694 - val_recall: 0.8442 - val_auc: 0.9739
Epoch 23/100
182276/182276 [==============================] - 1s 3us/sample - loss: 0.1782 - tp: 290.0000 - fp: 5589.0000 - tn: 176370.0000 - fn: 27.0000 - accuracy: 0.9692 - precision: 0.0493 - recall: 0.9148 - auc: 0.9768 - val_loss: 0.2565 - val_tp: 65.0000 - val_fp: 841.0000 - val_tn: 44651.0000 - val_fn: 12.0000 - val_accuracy: 0.9813 - val_precision: 0.0717 - val_recall: 0.8442 - val_auc: 0.9749
Epoch 24/100
182276/182276 [==============================] - 0s 3us/sample - loss: 0.2016 - tp: 288.0000 - fp: 5399.0000 - tn: 176560.0000 - fn: 29.0000 - accuracy: 0.9702 - precision: 0.0506 - recall: 0.9085 - auc: 0.9729 - val_loss: 0.2579 - val_tp: 65.0000 - val_fp: 778.0000 - val_tn: 44714.0000 - val_fn: 12.0000 - val_accuracy: 0.9827 - val_precision: 0.0771 - val_recall: 0.8442 - val_auc: 0.9748
Epoch 25/100
182276/182276 [==============================] - 0s 3us/sample - loss: 0.2121 - tp: 284.0000 - fp: 5515.0000 - tn: 176444.0000 - fn: 33.0000 - accuracy: 0.9696 - precision: 0.0490 - recall: 0.8959 - auc: 0.9713 - val_loss: 0.2525 - val_tp: 65.0000 - val_fp: 834.0000 - val_tn: 44658.0000 - val_fn: 12.0000 - val_accuracy: 0.9814 - val_precision: 0.0723 - val_recall: 0.8442 - val_auc: 0.9751
Epoch 26/100
182276/182276 [==============================] - 0s 3us/sample - loss: 0.2114 - tp: 283.0000 - fp: 5652.0000 - tn: 176307.0000 - fn: 34.0000 - accuracy: 0.9688 - precision: 0.0477 - recall: 0.8927 - auc: 0.9727 - val_loss: 0.2474 - val_tp: 65.0000 - val_fp: 884.0000 - val_tn: 44608.0000 - val_fn: 12.0000 - val_accuracy: 0.9803 - val_precision: 0.0685 - val_recall: 0.8442 - val_auc: 0.9755
Epoch 27/100
182276/182276 [==============================] - 0s 3us/sample - loss: 0.2202 - tp: 283.0000 - fp: 6261.0000 - tn: 175698.0000 - fn: 34.0000 - accuracy: 0.9655 - precision: 0.0432 - recall: 0.8927 - auc: 0.9678 - val_loss: 0.2424 - val_tp: 65.0000 - val_fp: 917.0000 - val_tn: 44575.0000 - val_fn: 12.0000 - val_accuracy: 0.9796 - val_precision: 0.0662 - val_recall: 0.8442 - val_auc: 0.9772
Epoch 28/100
182276/182276 [==============================] - 0s 3us/sample - loss: 0.2051 - tp: 281.0000 - fp: 6053.0000 - tn: 175906.0000 - fn: 36.0000 - accuracy: 0.9666 - precision: 0.0444 - recall: 0.8864 - auc: 0.9747 - val_loss: 0.2382 - val_tp: 65.0000 - val_fp: 966.0000 - val_tn: 44526.0000 - val_fn: 12.0000 - val_accuracy: 0.9785 - val_precision: 0.0630 - val_recall: 0.8442 - val_auc: 0.9772
Epoch 29/100
182276/182276 [==============================] - 0s 3us/sample - loss: 0.1738 - tp: 292.0000 - fp: 6087.0000 - tn: 175872.0000 - fn: 25.0000 - accuracy: 0.9665 - precision: 0.0458 - recall: 0.9211 - auc: 0.9786 - val_loss: 0.2402 - val_tp: 65.0000 - val_fp: 909.0000 - val_tn: 44583.0000 - val_fn: 12.0000 - val_accuracy: 0.9798 - val_precision: 0.0667 - val_recall: 0.8442 - val_auc: 0.9775
Epoch 30/100
182276/182276 [==============================] - 1s 3us/sample - loss: 0.1811 - tp: 290.0000 - fp: 5893.0000 - tn: 176066.0000 - fn: 27.0000 - accuracy: 0.9675 - precision: 0.0469 - recall: 0.9148 - auc: 0.9777 - val_loss: 0.2372 - val_tp: 65.0000 - val_fp: 898.0000 - val_tn: 44594.0000 - val_fn: 12.0000 - val_accuracy: 0.9800 - val_precision: 0.0675 - val_recall: 0.8442 - val_auc: 0.9779
Epoch 31/100
182276/182276 [==============================] - 0s 3us/sample - loss: 0.1971 - tp: 288.0000 - fp: 5681.0000 - tn: 176278.0000 - fn: 29.0000 - accuracy: 0.9687 - precision: 0.0482 - recall: 0.9085 - auc: 0.9732 - val_loss: 0.2387 - val_tp: 65.0000 - val_fp: 846.0000 - val_tn: 44646.0000 - val_fn: 12.0000 - val_accuracy: 0.9812 - val_precision: 0.0714 - val_recall: 0.8442 - val_auc: 0.9778
Epoch 32/100
182276/182276 [==============================] - 0s 3us/sample - loss: 0.1652 - tp: 293.0000 - fp: 5610.0000 - tn: 176349.0000 - fn: 24.0000 - accuracy: 0.9691 - precision: 0.0496 - recall: 0.9243 - auc: 0.9805 - val_loss: 0.2380 - val_tp: 65.0000 - val_fp: 810.0000 - val_tn: 44682.0000 - val_fn: 12.0000 - val_accuracy: 0.9820 - val_precision: 0.0743 - val_recall: 0.8442 - val_auc: 0.9792
Epoch 33/100
182276/182276 [==============================] - 0s 3us/sample - loss: 0.1743 - tp: 290.0000 - fp: 5337.0000 - tn: 176622.0000 - fn: 27.0000 - accuracy: 0.9706 - precision: 0.0515 - recall: 0.9148 - auc: 0.9812 - val_loss: 0.2379 - val_tp: 66.0000 - val_fp: 779.0000 - val_tn: 44713.0000 - val_fn: 11.0000 - val_accuracy: 0.9827 - val_precision: 0.0781 - val_recall: 0.8571 - val_auc: 0.9788
Epoch 34/100
182276/182276 [==============================] - 0s 3us/sample - loss: 0.1955 - tp: 282.0000 - fp: 5046.0000 - tn: 176913.0000 - fn: 35.0000 - accuracy: 0.9721 - precision: 0.0529 - recall: 0.8896 - auc: 0.9758 - val_loss: 0.2330 - val_tp: 66.0000 - val_fp: 810.0000 - val_tn: 44682.0000 - val_fn: 11.0000 - val_accuracy: 0.9820 - val_precision: 0.0753 - val_recall: 0.8571 - val_auc: 0.9789
Epoch 35/100
182276/182276 [==============================] - 0s 3us/sample - loss: 0.1800 - tp: 287.0000 - fp: 5462.0000 - tn: 176497.0000 - fn: 30.0000 - accuracy: 0.9699 - precision: 0.0499 - recall: 0.9054 - auc: 0.9792 - val_loss: 0.2311 - val_tp: 66.0000 - val_fp: 823.0000 - val_tn: 44669.0000 - val_fn: 11.0000 - val_accuracy: 0.9817 - val_precision: 0.0742 - val_recall: 0.8571 - val_auc: 0.9792
Epoch 36/100
182276/182276 [==============================] - 0s 3us/sample - loss: 0.1866 - tp: 287.0000 - fp: 5488.0000 - tn: 176471.0000 - fn: 30.0000 - accuracy: 0.9697 - precision: 0.0497 - recall: 0.9054 - auc: 0.9755 - val_loss: 0.2329 - val_tp: 66.0000 - val_fp: 822.0000 - val_tn: 44670.0000 - val_fn: 11.0000 - val_accuracy: 0.9817 - val_precision: 0.0743 - val_recall: 0.8571 - val_auc: 0.9794
Epoch 37/100
182276/182276 [==============================] - 0s 3us/sample - loss: 0.1948 - tp: 285.0000 - fp: 5659.0000 - tn: 176300.0000 - fn: 32.0000 - accuracy: 0.9688 - precision: 0.0479 - recall: 0.8991 - auc: 0.9764 - val_loss: 0.2299 - val_tp: 66.0000 - val_fp: 856.0000 - val_tn: 44636.0000 - val_fn: 11.0000 - val_accuracy: 0.9810 - val_precision: 0.0716 - val_recall: 0.8571 - val_auc: 0.9796
Epoch 38/100
182276/182276 [==============================] - 0s 3us/sample - loss: 0.1632 - tp: 295.0000 - fp: 5326.0000 - tn: 176633.0000 - fn: 22.0000 - accuracy: 0.9707 - precision: 0.0525 - recall: 0.9306 - auc: 0.9789 - val_loss: 0.2337 - val_tp: 66.0000 - val_fp: 777.0000 - val_tn: 44715.0000 - val_fn: 11.0000 - val_accuracy: 0.9827 - val_precision: 0.0783 - val_recall: 0.8571 - val_auc: 0.9796
Epoch 39/100
182276/182276 [==============================] - 0s 3us/sample - loss: 0.1646 - tp: 294.0000 - fp: 5457.0000 - tn: 176502.0000 - fn: 23.0000 - accuracy: 0.9699 - precision: 0.0511 - recall: 0.9274 - auc: 0.9812 - val_loss: 0.2332 - val_tp: 66.0000 - val_fp: 792.0000 - val_tn: 44700.0000 - val_fn: 11.0000 - val_accuracy: 0.9824 - val_precision: 0.0769 - val_recall: 0.8571 - val_auc: 0.9796
Epoch 40/100
182276/182276 [==============================] - 0s 3us/sample - loss: 0.1682 - tp: 292.0000 - fp: 4995.0000 - tn: 176964.0000 - fn: 25.0000 - accuracy: 0.9725 - precision: 0.0552 - recall: 0.9211 - auc: 0.9798 - val_loss: 0.2340 - val_tp: 66.0000 - val_fp: 754.0000 - val_tn: 44738.0000 - val_fn: 11.0000 - val_accuracy: 0.9832 - val_precision: 0.0805 - val_recall: 0.8571 - val_auc: 0.9794
Epoch 41/100
182276/182276 [==============================] - 0s 3us/sample - loss: 0.1827 - tp: 291.0000 - fp: 5188.0000 - tn: 176771.0000 - fn: 26.0000 - accuracy: 0.9714 - precision: 0.0531 - recall: 0.9180 - auc: 0.9737 - val_loss: 0.2349 - val_tp: 66.0000 - val_fp: 759.0000 - val_tn: 44733.0000 - val_fn: 11.0000 - val_accuracy: 0.9831 - val_precision: 0.0800 - val_recall: 0.8571 - val_auc: 0.9796
Epoch 42/100
182276/182276 [==============================] - 0s 3us/sample - loss: 0.1790 - tp: 291.0000 - fp: 4844.0000 - tn: 177115.0000 - fn: 26.0000 - accuracy: 0.9733 - precision: 0.0567 - recall: 0.9180 - auc: 0.9766 - val_loss: 0.2350 - val_tp: 66.0000 - val_fp: 749.0000 - val_tn: 44743.0000 - val_fn: 11.0000 - val_accuracy: 0.9833 - val_precision: 0.0810 - val_recall: 0.8571 - val_auc: 0.9792
Epoch 43/100
182276/182276 [==============================] - 1s 3us/sample - loss: 0.1884 - tp: 289.0000 - fp: 5341.0000 - tn: 176618.0000 - fn: 28.0000 - accuracy: 0.9705 - precision: 0.0513 - recall: 0.9117 - auc: 0.9745 - val_loss: 0.2303 - val_tp: 66.0000 - val_fp: 806.0000 - val_tn: 44686.0000 - val_fn: 11.0000 - val_accuracy: 0.9821 - val_precision: 0.0757 - val_recall: 0.8571 - val_auc: 0.9790
Epoch 44/100
182276/182276 [==============================] - 1s 3us/sample - loss: 0.1783 - tp: 289.0000 - fp: 5032.0000 - tn: 176927.0000 - fn: 28.0000 - accuracy: 0.9722 - precision: 0.0543 - recall: 0.9117 - auc: 0.9772 - val_loss: 0.2318 - val_tp: 66.0000 - val_fp: 755.0000 - val_tn: 44737.0000 - val_fn: 11.0000 - val_accuracy: 0.9832 - val_precision: 0.0804 - val_recall: 0.8571 - val_auc: 0.9792
Epoch 45/100
182276/182276 [==============================] - 0s 3us/sample - loss: 0.1410 - tp: 293.0000 - fp: 4829.0000 - tn: 177130.0000 - fn: 24.0000 - accuracy: 0.9734 - precision: 0.0572 - recall: 0.9243 - auc: 0.9863 - val_loss: 0.2316 - val_tp: 66.0000 - val_fp: 732.0000 - val_tn: 44760.0000 - val_fn: 11.0000 - val_accuracy: 0.9837 - val_precision: 0.0827 - val_recall: 0.8571 - val_auc: 0.9795
Epoch 46/100
182276/182276 [==============================] - 0s 3us/sample - loss: 0.1927 - tp: 284.0000 - fp: 5114.0000 - tn: 176845.0000 - fn: 33.0000 - accuracy: 0.9718 - precision: 0.0526 - recall: 0.8959 - auc: 0.9760 - val_loss: 0.2285 - val_tp: 66.0000 - val_fp: 731.0000 - val_tn: 44761.0000 - val_fn: 11.0000 - val_accuracy: 0.9837 - val_precision: 0.0828 - val_recall: 0.8571 - val_auc: 0.9808
Epoch 47/100
182276/182276 [==============================] - 0s 3us/sample - loss: 0.1594 - tp: 292.0000 - fp: 4918.0000 - tn: 177041.0000 - fn: 25.0000 - accuracy: 0.9729 - precision: 0.0560 - recall: 0.9211 - auc: 0.9836 - val_loss: 0.2272 - val_tp: 66.0000 - val_fp: 739.0000 - val_tn: 44753.0000 - val_fn: 11.0000 - val_accuracy: 0.9835 - val_precision: 0.0820 - val_recall: 0.8571 - val_auc: 0.9812
Epoch 48/100
182276/182276 [==============================] - 0s 3us/sample - loss: 0.1767 - tp: 287.0000 - fp: 4921.0000 - tn: 177038.0000 - fn: 30.0000 - accuracy: 0.9728 - precision: 0.0551 - recall: 0.9054 - auc: 0.9762 - val_loss: 0.2268 - val_tp: 66.0000 - val_fp: 740.0000 - val_tn: 44752.0000 - val_fn: 11.0000 - val_accuracy: 0.9835 - val_precision: 0.0819 - val_recall: 0.8571 - val_auc: 0.9812
Epoch 49/100
182276/182276 [==============================] - 0s 3us/sample - loss: 0.1636 - tp: 288.0000 - fp: 5131.0000 - tn: 176828.0000 - fn: 29.0000 - accuracy: 0.9717 - precision: 0.0531 - recall: 0.9085 - auc: 0.9833 - val_loss: 0.2255 - val_tp: 66.0000 - val_fp: 750.0000 - val_tn: 44742.0000 - val_fn: 11.0000 - val_accuracy: 0.9833 - val_precision: 0.0809 - val_recall: 0.8571 - val_auc: 0.9816
Epoch 50/100
182276/182276 [==============================] - 0s 3us/sample - loss: 0.1771 - tp: 288.0000 - fp: 5285.0000 - tn: 176674.0000 - fn: 29.0000 - accuracy: 0.9708 - precision: 0.0517 - recall: 0.9085 - auc: 0.9810 - val_loss: 0.2230 - val_tp: 66.0000 - val_fp: 807.0000 - val_tn: 44685.0000 - val_fn: 11.0000 - val_accuracy: 0.9820 - val_precision: 0.0756 - val_recall: 0.8571 - val_auc: 0.9816
Epoch 51/100
182276/182276 [==============================] - 1s 3us/sample - loss: 0.1516 - tp: 289.0000 - fp: 5492.0000 - tn: 176467.0000 - fn: 28.0000 - accuracy: 0.9697 - precision: 0.0500 - recall: 0.9117 - auc: 0.9868 - val_loss: 0.2204 - val_tp: 66.0000 - val_fp: 848.0000 - val_tn: 44644.0000 - val_fn: 11.0000 - val_accuracy: 0.9811 - val_precision: 0.0722 - val_recall: 0.8571 - val_auc: 0.9818
Epoch 52/100
182276/182276 [==============================] - 1s 3us/sample - loss: 0.1576 - tp: 293.0000 - fp: 5147.0000 - tn: 176812.0000 - fn: 24.0000 - accuracy: 0.9716 - precision: 0.0539 - recall: 0.9243 - auc: 0.9849 - val_loss: 0.2239 - val_tp: 66.0000 - val_fp: 749.0000 - val_tn: 44743.0000 - val_fn: 11.0000 - val_accuracy: 0.9833 - val_precision: 0.0810 - val_recall: 0.8571 - val_auc: 0.9820
Epoch 53/100
182276/182276 [==============================] - 0s 3us/sample - loss: 0.1611 - tp: 292.0000 - fp: 5003.0000 - tn: 176956.0000 - fn: 25.0000 - accuracy: 0.9724 - precision: 0.0551 - recall: 0.9211 - auc: 0.9797 - val_loss: 0.2263 - val_tp: 66.0000 - val_fp: 690.0000 - val_tn: 44802.0000 - val_fn: 11.0000 - val_accuracy: 0.9846 - val_precision: 0.0873 - val_recall: 0.8571 - val_auc: 0.9821
Epoch 54/100
182276/182276 [==============================] - 0s 3us/sample - loss: 0.1448 - tp: 296.0000 - fp: 4658.0000 - tn: 177301.0000 - fn: 21.0000 - accuracy: 0.9743 - precision: 0.0597 - recall: 0.9338 - auc: 0.9836 - val_loss: 0.2265 - val_tp: 66.0000 - val_fp: 673.0000 - val_tn: 44819.0000 - val_fn: 11.0000 - val_accuracy: 0.9850 - val_precision: 0.0893 - val_recall: 0.8571 - val_auc: 0.9821
Epoch 55/100
182276/182276 [==============================] - 0s 3us/sample - loss: 0.1661 - tp: 289.0000 - fp: 4662.0000 - tn: 177297.0000 - fn: 28.0000 - accuracy: 0.9743 - precision: 0.0584 - recall: 0.9117 - auc: 0.9809 - val_loss: 0.2250 - val_tp: 65.0000 - val_fp: 684.0000 - val_tn: 44808.0000 - val_fn: 12.0000 - val_accuracy: 0.9847 - val_precision: 0.0868 - val_recall: 0.8442 - val_auc: 0.9821
Epoch 56/100
182276/182276 [==============================] - 0s 3us/sample - loss: 0.1491 - tp: 290.0000 - fp: 4581.0000 - tn: 177378.0000 - fn: 27.0000 - accuracy: 0.9747 - precision: 0.0595 - recall: 0.9148 - auc: 0.9847 - val_loss: 0.2234 - val_tp: 66.0000 - val_fp: 708.0000 - val_tn: 44784.0000 - val_fn: 11.0000 - val_accuracy: 0.9842 - val_precision: 0.0853 - val_recall: 0.8571 - val_auc: 0.9815
Epoch 57/100
182276/182276 [==============================] - 0s 3us/sample - loss: 0.1317 - tp: 296.0000 - fp: 4693.0000 - tn: 177266.0000 - fn: 21.0000 - accuracy: 0.9741 - precision: 0.0593 - recall: 0.9338 - auc: 0.9890 - val_loss: 0.2252 - val_tp: 66.0000 - val_fp: 659.0000 - val_tn: 44833.0000 - val_fn: 11.0000 - val_accuracy: 0.9853 - val_precision: 0.0910 - val_recall: 0.8571 - val_auc: 0.9819
Epoch 58/100
182276/182276 [==============================] - 0s 3us/sample - loss: 0.1462 - tp: 293.0000 - fp: 4803.0000 - tn: 177156.0000 - fn: 24.0000 - accuracy: 0.9735 - precision: 0.0575 - recall: 0.9243 - auc: 0.9839 - val_loss: 0.2212 - val_tp: 66.0000 - val_fp: 710.0000 - val_tn: 44782.0000 - val_fn: 11.0000 - val_accuracy: 0.9842 - val_precision: 0.0851 - val_recall: 0.8571 - val_auc: 0.9822
Epoch 59/100
182276/182276 [==============================] - 0s 3us/sample - loss: 0.1572 - tp: 291.0000 - fp: 4882.0000 - tn: 177077.0000 - fn: 26.0000 - accuracy: 0.9731 - precision: 0.0563 - recall: 0.9180 - auc: 0.9835 - val_loss: 0.2209 - val_tp: 65.0000 - val_fp: 689.0000 - val_tn: 44803.0000 - val_fn: 12.0000 - val_accuracy: 0.9846 - val_precision: 0.0862 - val_recall: 0.8442 - val_auc: 0.9823
Epoch 60/100
182276/182276 [==============================] - 0s 3us/sample - loss: 0.1611 - tp: 292.0000 - fp: 4678.0000 - tn: 177281.0000 - fn: 25.0000 - accuracy: 0.9742 - precision: 0.0588 - recall: 0.9211 - auc: 0.9801 - val_loss: 0.2185 - val_tp: 66.0000 - val_fp: 705.0000 - val_tn: 44787.0000 - val_fn: 11.0000 - val_accuracy: 0.9843 - val_precision: 0.0856 - val_recall: 0.8571 - val_auc: 0.9824
Epoch 61/100
182276/182276 [==============================] - 0s 3us/sample - loss: 0.1478 - tp: 291.0000 - fp: 5050.0000 - tn: 176909.0000 - fn: 26.0000 - accuracy: 0.9722 - precision: 0.0545 - recall: 0.9180 - auc: 0.9848 - val_loss: 0.2179 - val_tp: 66.0000 - val_fp: 724.0000 - val_tn: 44768.0000 - val_fn: 11.0000 - val_accuracy: 0.9839 - val_precision: 0.0835 - val_recall: 0.8571 - val_auc: 0.9832
Epoch 62/100
182276/182276 [==============================] - 0s 3us/sample - loss: 0.1552 - tp: 293.0000 - fp: 4656.0000 - tn: 177303.0000 - fn: 24.0000 - accuracy: 0.9743 - precision: 0.0592 - recall: 0.9243 - auc: 0.9826 - val_loss: 0.2242 - val_tp: 65.0000 - val_fp: 611.0000 - val_tn: 44881.0000 - val_fn: 12.0000 - val_accuracy: 0.9863 - val_precision: 0.0962 - val_recall: 0.8442 - val_auc: 0.9825
Epoch 63/100
182276/182276 [==============================] - 0s 3us/sample - loss: 0.1609 - tp: 286.0000 - fp: 4642.0000 - tn: 177317.0000 - fn: 31.0000 - accuracy: 0.9744 - precision: 0.0580 - recall: 0.9022 - auc: 0.9850 - val_loss: 0.2184 - val_tp: 66.0000 - val_fp: 729.0000 - val_tn: 44763.0000 - val_fn: 11.0000 - val_accuracy: 0.9838 - val_precision: 0.0830 - val_recall: 0.8571 - val_auc: 0.9831
Epoch 64/100
182276/182276 [==============================] - 0s 3us/sample - loss: 0.1448 - tp: 292.0000 - fp: 4985.0000 - tn: 176974.0000 - fn: 25.0000 - accuracy: 0.9725 - precision: 0.0553 - recall: 0.9211 - auc: 0.9867 - val_loss: 0.2180 - val_tp: 65.0000 - val_fp: 743.0000 - val_tn: 44749.0000 - val_fn: 12.0000 - val_accuracy: 0.9834 - val_precision: 0.0804 - val_recall: 0.8442 - val_auc: 0.9831
Epoch 65/100
182276/182276 [==============================] - 0s 3us/sample - loss: 0.1498 - tp: 290.0000 - fp: 4687.0000 - tn: 177272.0000 - fn: 27.0000 - accuracy: 0.9741 - precision: 0.0583 - recall: 0.9148 - auc: 0.9851 - val_loss: 0.2216 - val_tp: 65.0000 - val_fp: 668.0000 - val_tn: 44824.0000 - val_fn: 12.0000 - val_accuracy: 0.9851 - val_precision: 0.0887 - val_recall: 0.8442 - val_auc: 0.9827
Epoch 66/100
182276/182276 [==============================] - 0s 3us/sample - loss: 0.1505 - tp: 292.0000 - fp: 4640.0000 - tn: 177319.0000 - fn: 25.0000 - accuracy: 0.9744 - precision: 0.0592 - recall: 0.9211 - auc: 0.9866 - val_loss: 0.2215 - val_tp: 65.0000 - val_fp: 674.0000 - val_tn: 44818.0000 - val_fn: 12.0000 - val_accuracy: 0.9849 - val_precision: 0.0880 - val_recall: 0.8442 - val_auc: 0.9829
Epoch 67/100
182276/182276 [==============================] - 0s 3us/sample - loss: 0.1527 - tp: 289.0000 - fp: 5125.0000 - tn: 176834.0000 - fn: 28.0000 - accuracy: 0.9717 - precision: 0.0534 - recall: 0.9117 - auc: 0.9852 - val_loss: 0.2158 - val_tp: 66.0000 - val_fp: 782.0000 - val_tn: 44710.0000 - val_fn: 11.0000 - val_accuracy: 0.9826 - val_precision: 0.0778 - val_recall: 0.8571 - val_auc: 0.9830
Epoch 68/100
182276/182276 [==============================] - 0s 3us/sample - loss: 0.1358 - tp: 292.0000 - fp: 5155.0000 - tn: 176804.0000 - fn: 25.0000 - accuracy: 0.9716 - precision: 0.0536 - recall: 0.9211 - auc: 0.9877 - val_loss: 0.2196 - val_tp: 66.0000 - val_fp: 742.0000 - val_tn: 44750.0000 - val_fn: 11.0000 - val_accuracy: 0.9835 - val_precision: 0.0817 - val_recall: 0.8571 - val_auc: 0.9827
Epoch 69/100
182276/182276 [==============================] - 0s 3us/sample - loss: 0.1553 - tp: 288.0000 - fp: 4992.0000 - tn: 176967.0000 - fn: 29.0000 - accuracy: 0.9725 - precision: 0.0545 - recall: 0.9085 - auc: 0.9858 - val_loss: 0.2191 - val_tp: 66.0000 - val_fp: 780.0000 - val_tn: 44712.0000 - val_fn: 11.0000 - val_accuracy: 0.9826 - val_precision: 0.0780 - val_recall: 0.8571 - val_auc: 0.9824
Epoch 70/100
182276/182276 [==============================] - 0s 3us/sample - loss: 0.1330 - tp: 295.0000 - fp: 5003.0000 - tn: 176956.0000 - fn: 22.0000 - accuracy: 0.9724 - precision: 0.0557 - recall: 0.9306 - auc: 0.9887 - val_loss: 0.2200 - val_tp: 66.0000 - val_fp: 716.0000 - val_tn: 44776.0000 - val_fn: 11.0000 - val_accuracy: 0.9840 - val_precision: 0.0844 - val_recall: 0.8571 - val_auc: 0.9827
Epoch 71/100
182276/182276 [==============================] - 0s 3us/sample - loss: 0.1510 - tp: 290.0000 - fp: 4960.0000 - tn: 176999.0000 - fn: 27.0000 - accuracy: 0.9726 - precision: 0.0552 - recall: 0.9148 - auc: 0.9853 - val_loss: 0.2170 - val_tp: 66.0000 - val_fp: 757.0000 - val_tn: 44735.0000 - val_fn: 11.0000 - val_accuracy: 0.9831 - val_precision: 0.0802 - val_recall: 0.8571 - val_auc: 0.9833
Epoch 72/100
182276/182276 [==============================] - 0s 3us/sample - loss: 0.1342 - tp: 293.0000 - fp: 4848.0000 - tn: 177111.0000 - fn: 24.0000 - accuracy: 0.9733 - precision: 0.0570 - recall: 0.9243 - auc: 0.9889 - val_loss: 0.2158 - val_tp: 66.0000 - val_fp: 776.0000 - val_tn: 44716.0000 - val_fn: 11.0000 - val_accuracy: 0.9827 - val_precision: 0.0784 - val_recall: 0.8571 - val_auc: 0.9829
Epoch 73/100
182276/182276 [==============================] - 0s 3us/sample - loss: 0.1492 - tp: 294.0000 - fp: 5065.0000 - tn: 176894.0000 - fn: 23.0000 - accuracy: 0.9721 - precision: 0.0549 - recall: 0.9274 - auc: 0.9829 - val_loss: 0.2188 - val_tp: 66.0000 - val_fp: 746.0000 - val_tn: 44746.0000 - val_fn: 11.0000 - val_accuracy: 0.9834 - val_precision: 0.0813 - val_recall: 0.8571 - val_auc: 0.9831
Epoch 74/100
182276/182276 [==============================] - 0s 3us/sample - loss: 0.1405 - tp: 294.0000 - fp: 5034.0000 - tn: 176925.0000 - fn: 23.0000 - accuracy: 0.9723 - precision: 0.0552 - recall: 0.9274 - auc: 0.9882 - val_loss: 0.2221 - val_tp: 66.0000 - val_fp: 703.0000 - val_tn: 44789.0000 - val_fn: 11.0000 - val_accuracy: 0.9843 - val_precision: 0.0858 - val_recall: 0.8571 - val_auc: 0.9820
Epoch 75/100
182276/182276 [==============================] - 0s 3us/sample - loss: 0.1292 - tp: 298.0000 - fp: 4958.0000 - tn: 177001.0000 - fn: 19.0000 - accuracy: 0.9727 - precision: 0.0567 - recall: 0.9401 - auc: 0.9895 - val_loss: 0.2209 - val_tp: 66.0000 - val_fp: 727.0000 - val_tn: 44765.0000 - val_fn: 11.0000 - val_accuracy: 0.9838 - val_precision: 0.0832 - val_recall: 0.8571 - val_auc: 0.9827
Epoch 76/100
182276/182276 [==============================] - 0s 3us/sample - loss: 0.1457 - tp: 291.0000 - fp: 4718.0000 - tn: 177241.0000 - fn: 26.0000 - accuracy: 0.9740 - precision: 0.0581 - recall: 0.9180 - auc: 0.9859 - val_loss: 0.2241 - val_tp: 66.0000 - val_fp: 688.0000 - val_tn: 44804.0000 - val_fn: 11.0000 - val_accuracy: 0.9847 - val_precision: 0.0875 - val_recall: 0.8571 - val_auc: 0.9828
Epoch 77/100
182276/182276 [==============================] - 0s 3us/sample - loss: 0.1491 - tp: 295.0000 - fp: 4661.0000 - tn: 177298.0000 - fn: 22.0000 - accuracy: 0.9743 - precision: 0.0595 - recall: 0.9306 - auc: 0.9855 - val_loss: 0.2215 - val_tp: 66.0000 - val_fp: 694.0000 - val_tn: 44798.0000 - val_fn: 11.0000 - val_accuracy: 0.9845 - val_precision: 0.0868 - val_recall: 0.8571 - val_auc: 0.9824
Epoch 78/100
182276/182276 [==============================] - 0s 3us/sample - loss: 0.1529 - tp: 295.0000 - fp: 4954.0000 - tn: 177005.0000 - fn: 22.0000 - accuracy: 0.9727 - precision: 0.0562 - recall: 0.9306 - auc: 0.9830 - val_loss: 0.2251 - val_tp: 66.0000 - val_fp: 704.0000 - val_tn: 44788.0000 - val_fn: 11.0000 - val_accuracy: 0.9843 - val_precision: 0.0857 - val_recall: 0.8571 - val_auc: 0.9823
Epoch 79/100
182276/182276 [==============================] - 0s 3us/sample - loss: 0.1302 - tp: 294.0000 - fp: 4550.0000 - tn: 177409.0000 - fn: 23.0000 - accuracy: 0.9749 - precision: 0.0607 - recall: 0.9274 - auc: 0.9891 - val_loss: 0.2273 - val_tp: 65.0000 - val_fp: 664.0000 - val_tn: 44828.0000 - val_fn: 12.0000 - val_accuracy: 0.9852 - val_precision: 0.0892 - val_recall: 0.8442 - val_auc: 0.9822
Epoch 80/100
182276/182276 [==============================] - 0s 3us/sample - loss: 0.1515 - tp: 288.0000 - fp: 4666.0000 - tn: 177293.0000 - fn: 29.0000 - accuracy: 0.9742 - precision: 0.0581 - recall: 0.9085 - auc: 0.9874 - val_loss: 0.2216 - val_tp: 66.0000 - val_fp: 721.0000 - val_tn: 44771.0000 - val_fn: 11.0000 - val_accuracy: 0.9839 - val_precision: 0.0839 - val_recall: 0.8571 - val_auc: 0.9828
Epoch 81/100
174080/182276 [===========================>..] - ETA: 0s - loss: 0.1249 - tp: 288.0000 - fp: 4817.0000 - tn: 168960.0000 - fn: 15.0000 - accuracy: 0.9722 - precision: 0.0564 - recall: 0.9505 - auc: 0.9885Restoring model weights from the end of the best epoch.
182276/182276 [==============================] - 0s 3us/sample - loss: 0.1258 - tp: 301.0000 - fp: 5042.0000 - tn: 176917.0000 - fn: 16.0000 - accuracy: 0.9723 - precision: 0.0563 - recall: 0.9495 - auc: 0.9885 - val_loss: 0.2229 - val_tp: 66.0000 - val_fp: 687.0000 - val_tn: 44805.0000 - val_fn: 11.0000 - val_accuracy: 0.9847 - val_precision: 0.0876 - val_recall: 0.8571 - val_auc: 0.9829
Epoch 00081: early stopping

Check training history

plot_metrics(weighted_history)

png

Evaluate metrics

train_predictions_weighted = weighted_model.predict(train_features, batch_size=BATCH_SIZE)
test_predictions_weighted = weighted_model.predict(test_features, batch_size=BATCH_SIZE)

weighted_results = weighted_model.evaluate(test_features, test_labels,
                                           batch_size=BATCH_SIZE, verbose=0)
for name, value in zip(weighted_model.metrics_names, weighted_results):
  print(name, ': ', value)
print()

plot_cm(test_labels, test_predictions_weighted)

loss :  0.07099118108907353
tp :  88.0
fp :  943.0
tn :  55921.0
fn :  10.0
accuracy :  0.9832696
precision :  0.08535402
recall :  0.8979592
auc :  0.99108195

Legitimate Transactions Detected (True Negatives):  55921
Legitimate Transactions Incorrectly Detected (False Positives):  943
Fraudulent Transactions Missed (False Negatives):  10
Fraudulent Transactions Detected (True Positives):  88
Total Fraudulent Transactions:  98

png

Here you can see that with class weights the accuracy and precision are lower because there are more false positives, but conversely the recall and AUC are higher because the model also found more true positives. Despite having lower accuracy, this model has higher recall (and identifies more fraudulent transactions). Of course, there is a cost to both types of error (you wouldn't want to bug users by flagging too many legitimate transactions as fraudulent, either). Carefully consider the trade offs between these different types of errors for your application.

Plot the ROC

plot_roc("Train Baseline", train_labels, train_predictions_baseline, color=colors[0])
plot_roc("Test Baseline", test_labels, test_predictions_baseline, color=colors[0], linestyle='--')

plot_roc("Train Weighted", train_labels, train_predictions_weighted, color=colors[1])
plot_roc("Test Weighted", test_labels, test_predictions_weighted, color=colors[1], linestyle='--')


plt.legend(loc='lower right')

<matplotlib.legend.Legend at 0x7ff23455eb38>

png

Oversampling

Oversample the minority class

A related approach would be to resample the dataset by oversampling the minority class.

pos_features = train_features[bool_train_labels]
neg_features = train_features[~bool_train_labels]

pos_labels = train_labels[bool_train_labels]
neg_labels = train_labels[~bool_train_labels]

Using NumPy

You can balance the dataset manually by choosing the right number of random indices from the positive examples:

ids = np.arange(len(pos_features))
choices = np.random.choice(ids, len(neg_features))

res_pos_features = pos_features[choices]
res_pos_labels = pos_labels[choices]

res_pos_features.shape

(181959, 29)
resampled_features = np.concatenate([res_pos_features, neg_features], axis=0)
resampled_labels = np.concatenate([res_pos_labels, neg_labels], axis=0)

order = np.arange(len(resampled_labels))
np.random.shuffle(order)
resampled_features = resampled_features[order]
resampled_labels = resampled_labels[order]

resampled_features.shape

(363918, 29)

Using tf.data

If you're using tf.data the easiest way to produce balanced examples is to start with a positive and a negative dataset, and merge them. See the tf.data guide for more examples.

BUFFER_SIZE = 100000

def make_ds(features, labels):
  ds = tf.data.Dataset.from_tensor_slices((features, labels))#.cache()
  ds = ds.shuffle(BUFFER_SIZE).repeat()
  return ds

pos_ds = make_ds(pos_features, pos_labels)
neg_ds = make_ds(neg_features, neg_labels)

Each dataset provides (feature, label) pairs:

for features, label in pos_ds.take(1):
  print("Features:\n", features.numpy())
  print()
  print("Label: ", label.numpy())

Features:
 [-3.03664638  3.22368542 -4.39930758  4.26241185 -1.02570116  0.50780559
 -3.40715885 -3.89448866 -0.9161103   0.41561662  4.33804249 -5.
 -1.18757492 -5.          1.09888547 -2.35998773 -3.86165628  0.14972806
  1.15212462 -1.45588855  5.         -2.06408405  2.17283177  0.23455764
  0.1493219   0.98286264  0.71376498  4.40248808  0.85020315]

Label:  1

Merge the two together using experimental.sample_from_datasets:

resampled_ds = tf.data.experimental.sample_from_datasets([pos_ds, neg_ds], weights=[0.5, 0.5])
resampled_ds = resampled_ds.batch(BATCH_SIZE).prefetch(2)

for features, label in resampled_ds.take(1):
  print(label.numpy().mean())

0.49658203125

To use this dataset, you'll need the number of steps per epoch.

The definition of "epoch" in this case is less clear. Say it's the number of batches required to see each negative example once:

resampled_steps_per_epoch = np.ceil(2.0*neg/BATCH_SIZE)
resampled_steps_per_epoch

278.0

Train on the oversampled data

Now try training the model with the resampled data set instead of using class weights to see how these methods compare.

resampled_model = make_model()
resampled_model.load_weights(initial_weights)

# Reset the bias to zero, since this dataset is balanced.
output_layer = resampled_model.layers[-1] 
output_layer.bias.assign([0])

val_ds = tf.data.Dataset.from_tensor_slices((val_features, val_labels)).cache()
val_ds = val_ds.batch(BATCH_SIZE).prefetch(2) 

resampled_history = resampled_model.fit(
    resampled_ds,
    epochs=EPOCHS,
    steps_per_epoch=resampled_steps_per_epoch,
    callbacks = [early_stopping],
    validation_data=val_ds)

Train for 278.0 steps, validate for 23 steps
Epoch 1/100
278/278 [==============================] - 9s 31ms/step - loss: 0.3993 - tp: 253951.0000 - fp: 80912.0000 - tn: 203506.0000 - fn: 30975.0000 - accuracy: 0.8035 - precision: 0.7584 - recall: 0.8913 - auc: 0.9170 - val_loss: 0.2451 - val_tp: 64.0000 - val_fp: 1245.0000 - val_tn: 44247.0000 - val_fn: 13.0000 - val_accuracy: 0.9724 - val_precision: 0.0489 - val_recall: 0.8312 - val_auc: 0.9512
Epoch 2/100
278/278 [==============================] - 6s 22ms/step - loss: 0.1993 - tp: 263820.0000 - fp: 17662.0000 - tn: 266682.0000 - fn: 21180.0000 - accuracy: 0.9318 - precision: 0.9373 - recall: 0.9257 - auc: 0.9754 - val_loss: 0.1296 - val_tp: 65.0000 - val_fp: 1007.0000 - val_tn: 44485.0000 - val_fn: 12.0000 - val_accuracy: 0.9776 - val_precision: 0.0606 - val_recall: 0.8442 - val_auc: 0.9623
Epoch 3/100
278/278 [==============================] - 6s 23ms/step - loss: 0.1556 - tp: 266494.0000 - fp: 11336.0000 - tn: 273535.0000 - fn: 17979.0000 - accuracy: 0.9485 - precision: 0.9592 - recall: 0.9368 - auc: 0.9846 - val_loss: 0.0933 - val_tp: 64.0000 - val_fp: 899.0000 - val_tn: 44593.0000 - val_fn: 13.0000 - val_accuracy: 0.9800 - val_precision: 0.0665 - val_recall: 0.8312 - val_auc: 0.9712
Epoch 4/100
278/278 [==============================] - 6s 23ms/step - loss: 0.1343 - tp: 268410.0000 - fp: 9529.0000 - tn: 274854.0000 - fn: 16551.0000 - accuracy: 0.9542 - precision: 0.9657 - recall: 0.9419 - auc: 0.9886 - val_loss: 0.0770 - val_tp: 65.0000 - val_fp: 820.0000 - val_tn: 44672.0000 - val_fn: 12.0000 - val_accuracy: 0.9817 - val_precision: 0.0734 - val_recall: 0.8442 - val_auc: 0.9714
Epoch 5/100
278/278 [==============================] - 6s 23ms/step - loss: 0.1204 - tp: 269261.0000 - fp: 8364.0000 - tn: 276457.0000 - fn: 15262.0000 - accuracy: 0.9585 - precision: 0.9699 - recall: 0.9464 - auc: 0.9912 - val_loss: 0.0676 - val_tp: 65.0000 - val_fp: 749.0000 - val_tn: 44743.0000 - val_fn: 12.0000 - val_accuracy: 0.9833 - val_precision: 0.0799 - val_recall: 0.8442 - val_auc: 0.9714
Epoch 6/100
278/278 [==============================] - 6s 23ms/step - loss: 0.1106 - tp: 269497.0000 - fp: 7589.0000 - tn: 277717.0000 - fn: 14541.0000 - accuracy: 0.9611 - precision: 0.9726 - recall: 0.9488 - auc: 0.9928 - val_loss: 0.0597 - val_tp: 65.0000 - val_fp: 674.0000 - val_tn: 44818.0000 - val_fn: 12.0000 - val_accuracy: 0.9849 - val_precision: 0.0880 - val_recall: 0.8442 - val_auc: 0.9711
Epoch 7/100
278/278 [==============================] - 6s 22ms/step - loss: 0.1027 - tp: 270278.0000 - fp: 7102.0000 - tn: 277756.0000 - fn: 14208.0000 - accuracy: 0.9626 - precision: 0.9744 - recall: 0.9501 - auc: 0.9940 - val_loss: 0.0545 - val_tp: 65.0000 - val_fp: 635.0000 - val_tn: 44857.0000 - val_fn: 12.0000 - val_accuracy: 0.9858 - val_precision: 0.0929 - val_recall: 0.8442 - val_auc: 0.9670
Epoch 8/100
278/278 [==============================] - 6s 23ms/step - loss: 0.0965 - tp: 271429.0000 - fp: 6755.0000 - tn: 277629.0000 - fn: 13531.0000 - accuracy: 0.9644 - precision: 0.9757 - recall: 0.9525 - auc: 0.9949 - val_loss: 0.0508 - val_tp: 65.0000 - val_fp: 599.0000 - val_tn: 44893.0000 - val_fn: 12.0000 - val_accuracy: 0.9866 - val_precision: 0.0979 - val_recall: 0.8442 - val_auc: 0.9641
Epoch 9/100
278/278 [==============================] - 6s 23ms/step - loss: 0.0910 - tp: 271471.0000 - fp: 6447.0000 - tn: 278236.0000 - fn: 13190.0000 - accuracy: 0.9655 - precision: 0.9768 - recall: 0.9537 - auc: 0.9955 - val_loss: 0.0453 - val_tp: 65.0000 - val_fp: 528.0000 - val_tn: 44964.0000 - val_fn: 12.0000 - val_accuracy: 0.9881 - val_precision: 0.1096 - val_recall: 0.8442 - val_auc: 0.9649
Epoch 10/100
278/278 [==============================] - 7s 24ms/step - loss: 0.0864 - tp: 271296.0000 - fp: 6076.0000 - tn: 279064.0000 - fn: 12908.0000 - accuracy: 0.9667 - precision: 0.9781 - recall: 0.9546 - auc: 0.9960 - val_loss: 0.0439 - val_tp: 65.0000 - val_fp: 527.0000 - val_tn: 44965.0000 - val_fn: 12.0000 - val_accuracy: 0.9882 - val_precision: 0.1098 - val_recall: 0.8442 - val_auc: 0.9663
Epoch 11/100
278/278 [==============================] - 7s 24ms/step - loss: 0.0837 - tp: 272018.0000 - fp: 5949.0000 - tn: 278669.0000 - fn: 12708.0000 - accuracy: 0.9672 - precision: 0.9786 - recall: 0.9554 - auc: 0.9963 - val_loss: 0.0403 - val_tp: 65.0000 - val_fp: 466.0000 - val_tn: 45026.0000 - val_fn: 12.0000 - val_accuracy: 0.9895 - val_precision: 0.1224 - val_recall: 0.8442 - val_auc: 0.9609
Epoch 12/100
278/278 [==============================] - 6s 22ms/step - loss: 0.0808 - tp: 272057.0000 - fp: 5717.0000 - tn: 279075.0000 - fn: 12495.0000 - accuracy: 0.9680 - precision: 0.9794 - recall: 0.9561 - auc: 0.9966 - val_loss: 0.0396 - val_tp: 65.0000 - val_fp: 466.0000 - val_tn: 45026.0000 - val_fn: 12.0000 - val_accuracy: 0.9895 - val_precision: 0.1224 - val_recall: 0.8442 - val_auc: 0.9620
Epoch 13/100
278/278 [==============================] - 6s 23ms/step - loss: 0.0781 - tp: 272148.0000 - fp: 5565.0000 - tn: 279525.0000 - fn: 12106.0000 - accuracy: 0.9690 - precision: 0.9800 - recall: 0.9574 - auc: 0.9968 - val_loss: 0.0370 - val_tp: 65.0000 - val_fp: 426.0000 - val_tn: 45066.0000 - val_fn: 12.0000 - val_accuracy: 0.9904 - val_precision: 0.1324 - val_recall: 0.8442 - val_auc: 0.9573
Epoch 14/100
278/278 [==============================] - 6s 23ms/step - loss: 0.0763 - tp: 273239.0000 - fp: 5582.0000 - tn: 278693.0000 - fn: 11830.0000 - accuracy: 0.9694 - precision: 0.9800 - recall: 0.9585 - auc: 0.9969 - val_loss: 0.0343 - val_tp: 65.0000 - val_fp: 387.0000 - val_tn: 45105.0000 - val_fn: 12.0000 - val_accuracy: 0.9912 - val_precision: 0.1438 - val_recall: 0.8442 - val_auc: 0.9469
Epoch 15/100
276/278 [============================>.] - ETA: 0s - loss: 0.0745 - tp: 271266.0000 - fp: 5297.0000 - tn: 277104.0000 - fn: 11581.0000 - accuracy: 0.9701 - precision: 0.9808 - recall: 0.9591 - auc: 0.9971Restoring model weights from the end of the best epoch.
278/278 [==============================] - 7s 24ms/step - loss: 0.0745 - tp: 273245.0000 - fp: 5329.0000 - tn: 279103.0000 - fn: 11667.0000 - accuracy: 0.9701 - precision: 0.9809 - recall: 0.9591 - auc: 0.9971 - val_loss: 0.0338 - val_tp: 65.0000 - val_fp: 385.0000 - val_tn: 45107.0000 - val_fn: 12.0000 - val_accuracy: 0.9913 - val_precision: 0.1444 - val_recall: 0.8442 - val_auc: 0.9483
Epoch 00015: early stopping

If the training process were considering the whole dataset on each gradient update, this oversampling would be basically identical to the class weighting.

But when training the model batch-wise, as you did here, the oversampled data provides a smoother gradient signal: Instead of each positive example being shown in one batch with a large weight, they're shown in many different batches each time with a small weight.

This smoother gradient signal makes it easier to train the model.

Check training history

Note that the distributions of metrics will be different here, because the training data has a totally different distribution from the validation and test data. 

plot_metrics(resampled_history )

png

Re-train

Because training is easier on the balanced data, the above training procedure may overfit quickly.

So break up the epochs to give the callbacks.EarlyStopping finer control over when to stop training.

resampled_model = make_model()
resampled_model.load_weights(initial_weights)

# Reset the bias to zero, since this dataset is balanced.
output_layer = resampled_model.layers[-1] 
output_layer.bias.assign([0])

resampled_history = resampled_model.fit(
    resampled_ds,
    # These are not real epochs
    steps_per_epoch = 20,
    epochs=10*EPOCHS,
    callbacks = [early_stopping],
    validation_data=(val_ds))

Train for 20 steps, validate for 23 steps
Epoch 1/1000
20/20 [==============================] - 2s 125ms/step - loss: 0.8089 - tp: 14347.0000 - fp: 12214.0000 - tn: 8299.0000 - fn: 6100.0000 - accuracy: 0.5529 - precision: 0.5402 - recall: 0.7017 - auc: 0.6092 - val_loss: 0.9011 - val_tp: 76.0000 - val_fp: 27659.0000 - val_tn: 17833.0000 - val_fn: 1.0000 - val_accuracy: 0.3930 - val_precision: 0.0027 - val_recall: 0.9870 - val_auc: 0.8595
Epoch 2/1000
20/20 [==============================] - 0s 19ms/step - loss: 0.6118 - tp: 17867.0000 - fp: 11296.0000 - tn: 9091.0000 - fn: 2706.0000 - accuracy: 0.6582 - precision: 0.6127 - recall: 0.8685 - auc: 0.8095 - val_loss: 0.7979 - val_tp: 75.0000 - val_fp: 24092.0000 - val_tn: 21400.0000 - val_fn: 2.0000 - val_accuracy: 0.4713 - val_precision: 0.0031 - val_recall: 0.9740 - val_auc: 0.9129
Epoch 3/1000
20/20 [==============================] - 0s 25ms/step - loss: 0.5263 - tp: 18444.0000 - fp: 9914.0000 - tn: 10542.0000 - fn: 2060.0000 - accuracy: 0.7077 - precision: 0.6504 - recall: 0.8995 - auc: 0.8731 - val_loss: 0.6968 - val_tp: 72.0000 - val_fp: 19388.0000 - val_tn: 26104.0000 - val_fn: 5.0000 - val_accuracy: 0.5744 - val_precision: 0.0037 - val_recall: 0.9351 - val_auc: 0.9243
Epoch 4/1000
20/20 [==============================] - 1s 26ms/step - loss: 0.4681 - tp: 18574.0000 - fp: 8511.0000 - tn: 11926.0000 - fn: 1949.0000 - accuracy: 0.7446 - precision: 0.6858 - recall: 0.9050 - auc: 0.9008 - val_loss: 0.6099 - val_tp: 69.0000 - val_fp: 14457.0000 - val_tn: 31035.0000 - val_fn: 8.0000 - val_accuracy: 0.6826 - val_precision: 0.0048 - val_recall: 0.8961 - val_auc: 0.9315
Epoch 5/1000
20/20 [==============================] - 0s 22ms/step - loss: 0.4214 - tp: 18741.0000 - fp: 7260.0000 - tn: 13095.0000 - fn: 1864.0000 - accuracy: 0.7772 - precision: 0.7208 - recall: 0.9095 - auc: 0.9183 - val_loss: 0.5397 - val_tp: 69.0000 - val_fp: 10027.0000 - val_tn: 35465.0000 - val_fn: 8.0000 - val_accuracy: 0.7798 - val_precision: 0.0068 - val_recall: 0.8961 - val_auc: 0.9365
Epoch 6/1000
20/20 [==============================] - 0s 24ms/step - loss: 0.3905 - tp: 18439.0000 - fp: 6217.0000 - tn: 14449.0000 - fn: 1855.0000 - accuracy: 0.8029 - precision: 0.7479 - recall: 0.9086 - auc: 0.9278 - val_loss: 0.4800 - val_tp: 68.0000 - val_fp: 6291.0000 - val_tn: 39201.0000 - val_fn: 9.0000 - val_accuracy: 0.8617 - val_precision: 0.0107 - val_recall: 0.8831 - val_auc: 0.9401
Epoch 7/1000
20/20 [==============================] - 0s 22ms/step - loss: 0.3582 - tp: 18574.0000 - fp: 5184.0000 - tn: 15312.0000 - fn: 1890.0000 - accuracy: 0.8273 - precision: 0.7818 - recall: 0.9076 - auc: 0.9361 - val_loss: 0.4308 - val_tp: 66.0000 - val_fp: 4109.0000 - val_tn: 41383.0000 - val_fn: 11.0000 - val_accuracy: 0.9096 - val_precision: 0.0158 - val_recall: 0.8571 - val_auc: 0.9435
Epoch 8/1000
20/20 [==============================] - 0s 21ms/step - loss: 0.3304 - tp: 18640.0000 - fp: 4273.0000 - tn: 16202.0000 - fn: 1845.0000 - accuracy: 0.8506 - precision: 0.8135 - recall: 0.9099 - auc: 0.9439 - val_loss: 0.3901 - val_tp: 66.0000 - val_fp: 2897.0000 - val_tn: 42595.0000 - val_fn: 11.0000 - val_accuracy: 0.9362 - val_precision: 0.0223 - val_recall: 0.8571 - val_auc: 0.9457
Epoch 9/1000
20/20 [==============================] - 1s 29ms/step - loss: 0.3117 - tp: 18608.0000 - fp: 3690.0000 - tn: 16838.0000 - fn: 1824.0000 - accuracy: 0.8654 - precision: 0.8345 - recall: 0.9107 - auc: 0.9494 - val_loss: 0.3563 - val_tp: 65.0000 - val_fp: 2255.0000 - val_tn: 43237.0000 - val_fn: 12.0000 - val_accuracy: 0.9503 - val_precision: 0.0280 - val_recall: 0.8442 - val_auc: 0.9471
Epoch 10/1000
20/20 [==============================] - 0s 24ms/step - loss: 0.2959 - tp: 18705.0000 - fp: 3143.0000 - tn: 17316.0000 - fn: 1796.0000 - accuracy: 0.8794 - precision: 0.8561 - recall: 0.9124 - auc: 0.9526 - val_loss: 0.3266 - val_tp: 65.0000 - val_fp: 1885.0000 - val_tn: 43607.0000 - val_fn: 12.0000 - val_accuracy: 0.9584 - val_precision: 0.0333 - val_recall: 0.8442 - val_auc: 0.9480
Epoch 11/1000
20/20 [==============================] - 0s 24ms/step - loss: 0.2792 - tp: 18667.0000 - fp: 2780.0000 - tn: 17694.0000 - fn: 1819.0000 - accuracy: 0.8877 - precision: 0.8704 - recall: 0.9112 - auc: 0.9567 - val_loss: 0.3012 - val_tp: 65.0000 - val_fp: 1653.0000 - val_tn: 43839.0000 - val_fn: 12.0000 - val_accuracy: 0.9635 - val_precision: 0.0378 - val_recall: 0.8442 - val_auc: 0.9490
Epoch 12/1000
20/20 [==============================] - 1s 27ms/step - loss: 0.2657 - tp: 18875.0000 - fp: 2429.0000 - tn: 17911.0000 - fn: 1745.0000 - accuracy: 0.8981 - precision: 0.8860 - recall: 0.9154 - auc: 0.9605 - val_loss: 0.2796 - val_tp: 65.0000 - val_fp: 1479.0000 - val_tn: 44013.0000 - val_fn: 12.0000 - val_accuracy: 0.9673 - val_precision: 0.0421 - val_recall: 0.8442 - val_auc: 0.9498
Epoch 13/1000
20/20 [==============================] - 0s 25ms/step - loss: 0.2525 - tp: 18848.0000 - fp: 2209.0000 - tn: 18220.0000 - fn: 1683.0000 - accuracy: 0.9050 - precision: 0.8951 - recall: 0.9180 - auc: 0.9640 - val_loss: 0.2601 - val_tp: 64.0000 - val_fp: 1360.0000 - val_tn: 44132.0000 - val_fn: 13.0000 - val_accuracy: 0.9699 - val_precision: 0.0449 - val_recall: 0.8312 - val_auc: 0.9500
Epoch 14/1000
20/20 [==============================] - 0s 21ms/step - loss: 0.2415 - tp: 18810.0000 - fp: 1926.0000 - tn: 18558.0000 - fn: 1666.0000 - accuracy: 0.9123 - precision: 0.9071 - recall: 0.9186 - auc: 0.9666 - val_loss: 0.2432 - val_tp: 64.0000 - val_fp: 1262.0000 - val_tn: 44230.0000 - val_fn: 13.0000 - val_accuracy: 0.9720 - val_precision: 0.0483 - val_recall: 0.8312 - val_auc: 0.9513
Epoch 15/1000
20/20 [==============================] - 0s 21ms/step - loss: 0.2354 - tp: 18817.0000 - fp: 1817.0000 - tn: 18689.0000 - fn: 1637.0000 - accuracy: 0.9157 - precision: 0.9119 - recall: 0.9200 - auc: 0.9673 - val_loss: 0.2274 - val_tp: 64.0000 - val_fp: 1185.0000 - val_tn: 44307.0000 - val_fn: 13.0000 - val_accuracy: 0.9737 - val_precision: 0.0512 - val_recall: 0.8312 - val_auc: 0.9519
Epoch 16/1000
20/20 [==============================] - 0s 21ms/step - loss: 0.2267 - tp: 18766.0000 - fp: 1715.0000 - tn: 18892.0000 - fn: 1587.0000 - accuracy: 0.9194 - precision: 0.9163 - recall: 0.9220 - auc: 0.9698 - val_loss: 0.2130 - val_tp: 64.0000 - val_fp: 1123.0000 - val_tn: 44369.0000 - val_fn: 13.0000 - val_accuracy: 0.9751 - val_precision: 0.0539 - val_recall: 0.8312 - val_auc: 0.9524
Epoch 17/1000
20/20 [==============================] - 0s 21ms/step - loss: 0.2216 - tp: 18903.0000 - fp: 1530.0000 - tn: 18898.0000 - fn: 1629.0000 - accuracy: 0.9229 - precision: 0.9251 - recall: 0.9207 - auc: 0.9701 - val_loss: 0.2008 - val_tp: 64.0000 - val_fp: 1091.0000 - val_tn: 44401.0000 - val_fn: 13.0000 - val_accuracy: 0.9758 - val_precision: 0.0554 - val_recall: 0.8312 - val_auc: 0.9537
Epoch 18/1000
20/20 [==============================] - 1s 29ms/step - loss: 0.2127 - tp: 18973.0000 - fp: 1436.0000 - tn: 18972.0000 - fn: 1579.0000 - accuracy: 0.9264 - precision: 0.9296 - recall: 0.9232 - auc: 0.9727 - val_loss: 0.1900 - val_tp: 64.0000 - val_fp: 1055.0000 - val_tn: 44437.0000 - val_fn: 13.0000 - val_accuracy: 0.9766 - val_precision: 0.0572 - val_recall: 0.8312 - val_auc: 0.9546
Epoch 19/1000
20/20 [==============================] - 0s 23ms/step - loss: 0.2091 - tp: 18809.0000 - fp: 1325.0000 - tn: 19269.0000 - fn: 1557.0000 - accuracy: 0.9296 - precision: 0.9342 - recall: 0.9235 - auc: 0.9733 - val_loss: 0.1799 - val_tp: 64.0000 - val_fp: 1013.0000 - val_tn: 44479.0000 - val_fn: 13.0000 - val_accuracy: 0.9775 - val_precision: 0.0594 - val_recall: 0.8312 - val_auc: 0.9551
Epoch 20/1000
20/20 [==============================] - 0s 24ms/step - loss: 0.2023 - tp: 19032.0000 - fp: 1268.0000 - tn: 19089.0000 - fn: 1571.0000 - accuracy: 0.9307 - precision: 0.9375 - recall: 0.9237 - auc: 0.9747 - val_loss: 0.1716 - val_tp: 64.0000 - val_fp: 995.0000 - val_tn: 44497.0000 - val_fn: 13.0000 - val_accuracy: 0.9779 - val_precision: 0.0604 - val_recall: 0.8312 - val_auc: 0.9560
Epoch 21/1000
20/20 [==============================] - 0s 25ms/step - loss: 0.2018 - tp: 18748.0000 - fp: 1267.0000 - tn: 19418.0000 - fn: 1527.0000 - accuracy: 0.9318 - precision: 0.9367 - recall: 0.9247 - auc: 0.9749 - val_loss: 0.1640 - val_tp: 64.0000 - val_fp: 984.0000 - val_tn: 44508.0000 - val_fn: 13.0000 - val_accuracy: 0.9781 - val_precision: 0.0611 - val_recall: 0.8312 - val_auc: 0.9568
Epoch 22/1000
20/20 [==============================] - 1s 26ms/step - loss: 0.1896 - tp: 19105.0000 - fp: 1122.0000 - tn: 19262.0000 - fn: 1471.0000 - accuracy: 0.9367 - precision: 0.9445 - recall: 0.9285 - auc: 0.9779 - val_loss: 0.1580 - val_tp: 64.0000 - val_fp: 1014.0000 - val_tn: 44478.0000 - val_fn: 13.0000 - val_accuracy: 0.9775 - val_precision: 0.0594 - val_recall: 0.8312 - val_auc: 0.9574
Epoch 23/1000
20/20 [==============================] - 0s 22ms/step - loss: 0.1872 - tp: 19153.0000 - fp: 1123.0000 - tn: 19211.0000 - fn: 1473.0000 - accuracy: 0.9366 - precision: 0.9446 - recall: 0.9286 - auc: 0.9779 - val_loss: 0.1526 - val_tp: 64.0000 - val_fp: 1024.0000 - val_tn: 44468.0000 - val_fn: 13.0000 - val_accuracy: 0.9772 - val_precision: 0.0588 - val_recall: 0.8312 - val_auc: 0.9581
Epoch 24/1000
20/20 [==============================] - 0s 23ms/step - loss: 0.1843 - tp: 19073.0000 - fp: 1138.0000 - tn: 19330.0000 - fn: 1419.0000 - accuracy: 0.9376 - precision: 0.9437 - recall: 0.9308 - auc: 0.9790 - val_loss: 0.1465 - val_tp: 64.0000 - val_fp: 999.0000 - val_tn: 44493.0000 - val_fn: 13.0000 - val_accuracy: 0.9778 - val_precision: 0.0602 - val_recall: 0.8312 - val_auc: 0.9589
Epoch 25/1000
20/20 [==============================] - 0s 22ms/step - loss: 0.1787 - tp: 19103.0000 - fp: 1006.0000 - tn: 19412.0000 - fn: 1439.0000 - accuracy: 0.9403 - precision: 0.9500 - recall: 0.9299 - auc: 0.9800 - val_loss: 0.1413 - val_tp: 64.0000 - val_fp: 994.0000 - val_tn: 44498.0000 - val_fn: 13.0000 - val_accuracy: 0.9779 - val_precision: 0.0605 - val_recall: 0.8312 - val_auc: 0.9599
Epoch 26/1000
20/20 [==============================] - 0s 21ms/step - loss: 0.1785 - tp: 19169.0000 - fp: 1019.0000 - tn: 19333.0000 - fn: 1439.0000 - accuracy: 0.9400 - precision: 0.9495 - recall: 0.9302 - auc: 0.9800 - val_loss: 0.1370 - val_tp: 64.0000 - val_fp: 998.0000 - val_tn: 44494.0000 - val_fn: 13.0000 - val_accuracy: 0.9778 - val_precision: 0.0603 - val_recall: 0.8312 - val_auc: 0.9612
Epoch 27/1000
20/20 [==============================] - 1s 28ms/step - loss: 0.1728 - tp: 19022.0000 - fp: 967.0000 - tn: 19608.0000 - fn: 1363.0000 - accuracy: 0.9431 - precision: 0.9516 - recall: 0.9331 - auc: 0.9812 - val_loss: 0.1323 - val_tp: 64.0000 - val_fp: 991.0000 - val_tn: 44501.0000 - val_fn: 13.0000 - val_accuracy: 0.9780 - val_precision: 0.0607 - val_recall: 0.8312 - val_auc: 0.9625
Epoch 28/1000
20/20 [==============================] - 1s 25ms/step - loss: 0.1712 - tp: 19308.0000 - fp: 954.0000 - tn: 19330.0000 - fn: 1368.0000 - accuracy: 0.9433 - precision: 0.9529 - recall: 0.9338 - auc: 0.9819 - val_loss: 0.1276 - val_tp: 64.0000 - val_fp: 966.0000 - val_tn: 44526.0000 - val_fn: 13.0000 - val_accuracy: 0.9785 - val_precision: 0.0621 - val_recall: 0.8312 - val_auc: 0.9634
Epoch 29/1000
20/20 [==============================] - 1s 25ms/step - loss: 0.1708 - tp: 19156.0000 - fp: 938.0000 - tn: 19497.0000 - fn: 1369.0000 - accuracy: 0.9437 - precision: 0.9533 - recall: 0.9333 - auc: 0.9814 - val_loss: 0.1233 - val_tp: 64.0000 - val_fp: 943.0000 - val_tn: 44549.0000 - val_fn: 13.0000 - val_accuracy: 0.9790 - val_precision: 0.0636 - val_recall: 0.8312 - val_auc: 0.9642
Epoch 30/1000
20/20 [==============================] - 1s 26ms/step - loss: 0.1665 - tp: 19082.0000 - fp: 901.0000 - tn: 19620.0000 - fn: 1357.0000 - accuracy: 0.9449 - precision: 0.9549 - recall: 0.9336 - auc: 0.9826 - val_loss: 0.1203 - val_tp: 64.0000 - val_fp: 950.0000 - val_tn: 44542.0000 - val_fn: 13.0000 - val_accuracy: 0.9789 - val_precision: 0.0631 - val_recall: 0.8312 - val_auc: 0.9651
Epoch 31/1000
20/20 [==============================] - 0s 24ms/step - loss: 0.1625 - tp: 19249.0000 - fp: 859.0000 - tn: 19466.0000 - fn: 1386.0000 - accuracy: 0.9452 - precision: 0.9573 - recall: 0.9328 - auc: 0.9831 - val_loss: 0.1177 - val_tp: 64.0000 - val_fp: 960.0000 - val_tn: 44532.0000 - val_fn: 13.0000 - val_accuracy: 0.9786 - val_precision: 0.0625 - val_recall: 0.8312 - val_auc: 0.9662
Epoch 32/1000
20/20 [==============================] - 0s 22ms/step - loss: 0.1632 - tp: 19227.0000 - fp: 861.0000 - tn: 19515.0000 - fn: 1357.0000 - accuracy: 0.9458 - precision: 0.9571 - recall: 0.9341 - auc: 0.9832 - val_loss: 0.1152 - val_tp: 64.0000 - val_fp: 968.0000 - val_tn: 44524.0000 - val_fn: 13.0000 - val_accuracy: 0.9785 - val_precision: 0.0620 - val_recall: 0.8312 - val_auc: 0.9671
Epoch 33/1000
20/20 [==============================] - 0s 22ms/step - loss: 0.1578 - tp: 19253.0000 - fp: 807.0000 - tn: 19595.0000 - fn: 1305.0000 - accuracy: 0.9484 - precision: 0.9598 - recall: 0.9365 - auc: 0.9841 - val_loss: 0.1123 - val_tp: 64.0000 - val_fp: 966.0000 - val_tn: 44526.0000 - val_fn: 13.0000 - val_accuracy: 0.9785 - val_precision: 0.0621 - val_recall: 0.8312 - val_auc: 0.9680
Epoch 34/1000
20/20 [==============================] - 0s 21ms/step - loss: 0.1601 - tp: 19074.0000 - fp: 912.0000 - tn: 19671.0000 - fn: 1303.0000 - accuracy: 0.9459 - precision: 0.9544 - recall: 0.9361 - auc: 0.9838 - val_loss: 0.1089 - val_tp: 64.0000 - val_fp: 946.0000 - val_tn: 44546.0000 - val_fn: 13.0000 - val_accuracy: 0.9790 - val_precision: 0.0634 - val_recall: 0.8312 - val_auc: 0.9677
Epoch 35/1000
20/20 [==============================] - 0s 24ms/step - loss: 0.1540 - tp: 19203.0000 - fp: 774.0000 - tn: 19677.0000 - fn: 1306.0000 - accuracy: 0.9492 - precision: 0.9613 - recall: 0.9363 - auc: 0.9848 - val_loss: 0.1068 - val_tp: 64.0000 - val_fp: 948.0000 - val_tn: 44544.0000 - val_fn: 13.0000 - val_accuracy: 0.9789 - val_precision: 0.0632 - val_recall: 0.8312 - val_auc: 0.9687
Epoch 36/1000
20/20 [==============================] - 1s 30ms/step - loss: 0.1531 - tp: 19289.0000 - fp: 839.0000 - tn: 19554.0000 - fn: 1278.0000 - accuracy: 0.9483 - precision: 0.9583 - recall: 0.9379 - auc: 0.9851 - val_loss: 0.1049 - val_tp: 64.0000 - val_fp: 952.0000 - val_tn: 44540.0000 - val_fn: 13.0000 - val_accuracy: 0.9788 - val_precision: 0.0630 - val_recall: 0.8312 - val_auc: 0.9693
Epoch 37/1000
20/20 [==============================] - 0s 24ms/step - loss: 0.1520 - tp: 19159.0000 - fp: 787.0000 - tn: 19740.0000 - fn: 1274.0000 - accuracy: 0.9497 - precision: 0.9605 - recall: 0.9376 - auc: 0.9854 - val_loss: 0.1016 - val_tp: 64.0000 - val_fp: 903.0000 - val_tn: 44589.0000 - val_fn: 13.0000 - val_accuracy: 0.9799 - val_precision: 0.0662 - val_recall: 0.8312 - val_auc: 0.9691
Epoch 38/1000
20/20 [==============================] - 0s 24ms/step - loss: 0.1503 - tp: 19088.0000 - fp: 783.0000 - tn: 19822.0000 - fn: 1267.0000 - accuracy: 0.9500 - precision: 0.9606 - recall: 0.9378 - auc: 0.9854 - val_loss: 0.0986 - val_tp: 64.0000 - val_fp: 867.0000 - val_tn: 44625.0000 - val_fn: 13.0000 - val_accuracy: 0.9807 - val_precision: 0.0687 - val_recall: 0.8312 - val_auc: 0.9698
Epoch 39/1000
20/20 [==============================] - 0s 24ms/step - loss: 0.1461 - tp: 18963.0000 - fp: 697.0000 - tn: 20058.0000 - fn: 1242.0000 - accuracy: 0.9527 - precision: 0.9645 - recall: 0.9385 - auc: 0.9864 - val_loss: 0.0968 - val_tp: 64.0000 - val_fp: 873.0000 - val_tn: 44619.0000 - val_fn: 13.0000 - val_accuracy: 0.9806 - val_precision: 0.0683 - val_recall: 0.8312 - val_auc: 0.9704
Epoch 40/1000
20/20 [==============================] - 0s 25ms/step - loss: 0.1488 - tp: 19182.0000 - fp: 765.0000 - tn: 19753.0000 - fn: 1260.0000 - accuracy: 0.9506 - precision: 0.9616 - recall: 0.9384 - auc: 0.9858 - val_loss: 0.0952 - val_tp: 64.0000 - val_fp: 888.0000 - val_tn: 44604.0000 - val_fn: 13.0000 - val_accuracy: 0.9802 - val_precision: 0.0672 - val_recall: 0.8312 - val_auc: 0.9709
Epoch 41/1000
20/20 [==============================] - 0s 21ms/step - loss: 0.1461 - tp: 19216.0000 - fp: 731.0000 - tn: 19760.0000 - fn: 1253.0000 - accuracy: 0.9516 - precision: 0.9634 - recall: 0.9388 - auc: 0.9864 - val_loss: 0.0940 - val_tp: 64.0000 - val_fp: 900.0000 - val_tn: 44592.0000 - val_fn: 13.0000 - val_accuracy: 0.9800 - val_precision: 0.0664 - val_recall: 0.8312 - val_auc: 0.9702
Epoch 42/1000
20/20 [==============================] - 0s 22ms/step - loss: 0.1454 - tp: 19271.0000 - fp: 739.0000 - tn: 19725.0000 - fn: 1225.0000 - accuracy: 0.9521 - precision: 0.9631 - recall: 0.9402 - auc: 0.9866 - val_loss: 0.0922 - val_tp: 65.0000 - val_fp: 891.0000 - val_tn: 44601.0000 - val_fn: 12.0000 - val_accuracy: 0.9802 - val_precision: 0.0680 - val_recall: 0.8442 - val_auc: 0.9704
Epoch 43/1000
20/20 [==============================] - 0s 23ms/step - loss: 0.1408 - tp: 19367.0000 - fp: 741.0000 - tn: 19635.0000 - fn: 1217.0000 - accuracy: 0.9522 - precision: 0.9631 - recall: 0.9409 - auc: 0.9874 - val_loss: 0.0906 - val_tp: 64.0000 - val_fp: 887.0000 - val_tn: 44605.0000 - val_fn: 13.0000 - val_accuracy: 0.9802 - val_precision: 0.0673 - val_recall: 0.8312 - val_auc: 0.9710
Epoch 44/1000
20/20 [==============================] - 0s 22ms/step - loss: 0.1393 - tp: 19044.0000 - fp: 687.0000 - tn: 20017.0000 - fn: 1212.0000 - accuracy: 0.9536 - precision: 0.9652 - recall: 0.9402 - auc: 0.9876 - val_loss: 0.0886 - val_tp: 64.0000 - val_fp: 867.0000 - val_tn: 44625.0000 - val_fn: 13.0000 - val_accuracy: 0.9807 - val_precision: 0.0687 - val_recall: 0.8312 - val_auc: 0.9707
Epoch 45/1000
20/20 [==============================] - 1s 29ms/step - loss: 0.1380 - tp: 19394.0000 - fp: 709.0000 - tn: 19660.0000 - fn: 1197.0000 - accuracy: 0.9535 - precision: 0.9647 - recall: 0.9419 - auc: 0.9878 - val_loss: 0.0876 - val_tp: 64.0000 - val_fp: 869.0000 - val_tn: 44623.0000 - val_fn: 13.0000 - val_accuracy: 0.9806 - val_precision: 0.0686 - val_recall: 0.8312 - val_auc: 0.9712
Epoch 46/1000
20/20 [==============================] - 1s 25ms/step - loss: 0.1404 - tp: 19312.0000 - fp: 721.0000 - tn: 19701.0000 - fn: 1226.0000 - accuracy: 0.9525 - precision: 0.9640 - recall: 0.9403 - auc: 0.9875 - val_loss: 0.0869 - val_tp: 64.0000 - val_fp: 868.0000 - val_tn: 44624.0000 - val_fn: 13.0000 - val_accuracy: 0.9807 - val_precision: 0.0687 - val_recall: 0.8312 - val_auc: 0.9716
Epoch 47/1000
20/20 [==============================] - 1s 25ms/step - loss: 0.1340 - tp: 19559.0000 - fp: 629.0000 - tn: 19603.0000 - fn: 1169.0000 - accuracy: 0.9561 - precision: 0.9688 - recall: 0.9436 - auc: 0.9887 - val_loss: 0.0857 - val_tp: 64.0000 - val_fp: 866.0000 - val_tn: 44626.0000 - val_fn: 13.0000 - val_accuracy: 0.9807 - val_precision: 0.0688 - val_recall: 0.8312 - val_auc: 0.9718
Epoch 48/1000
20/20 [==============================] - 1s 25ms/step - loss: 0.1362 - tp: 19326.0000 - fp: 712.0000 - tn: 19759.0000 - fn: 1163.0000 - accuracy: 0.9542 - precision: 0.9645 - recall: 0.9432 - auc: 0.9883 - val_loss: 0.0843 - val_tp: 64.0000 - val_fp: 856.0000 - val_tn: 44636.0000 - val_fn: 13.0000 - val_accuracy: 0.9809 - val_precision: 0.0696 - val_recall: 0.8312 - val_auc: 0.9715
Epoch 49/1000
20/20 [==============================] - 0s 23ms/step - loss: 0.1311 - tp: 19406.0000 - fp: 594.0000 - tn: 19798.0000 - fn: 1162.0000 - accuracy: 0.9571 - precision: 0.9703 - recall: 0.9435 - auc: 0.9890 - val_loss: 0.0843 - val_tp: 64.0000 - val_fp: 883.0000 - val_tn: 44609.0000 - val_fn: 13.0000 - val_accuracy: 0.9803 - val_precision: 0.0676 - val_recall: 0.8312 - val_auc: 0.9720
Epoch 50/1000
20/20 [==============================] - 0s 21ms/step - loss: 0.1327 - tp: 19407.0000 - fp: 698.0000 - tn: 19672.0000 - fn: 1183.0000 - accuracy: 0.9541 - precision: 0.9653 - recall: 0.9425 - auc: 0.9887 - val_loss: 0.0838 - val_tp: 64.0000 - val_fp: 896.0000 - val_tn: 44596.0000 - val_fn: 13.0000 - val_accuracy: 0.9801 - val_precision: 0.0667 - val_recall: 0.8312 - val_auc: 0.9721
Epoch 51/1000
20/20 [==============================] - 0s 23ms/step - loss: 0.1326 - tp: 19092.0000 - fp: 673.0000 - tn: 20067.0000 - fn: 1128.0000 - accuracy: 0.9560 - precision: 0.9659 - recall: 0.9442 - auc: 0.9890 - val_loss: 0.0821 - val_tp: 64.0000 - val_fp: 874.0000 - val_tn: 44618.0000 - val_fn: 13.0000 - val_accuracy: 0.9805 - val_precision: 0.0682 - val_recall: 0.8312 - val_auc: 0.9718
Epoch 52/1000
20/20 [==============================] - 0s 22ms/step - loss: 0.1295 - tp: 19232.0000 - fp: 664.0000 - tn: 19899.0000 - fn: 1165.0000 - accuracy: 0.9553 - precision: 0.9666 - recall: 0.9429 - auc: 0.9894 - val_loss: 0.0802 - val_tp: 64.0000 - val_fp: 843.0000 - val_tn: 44649.0000 - val_fn: 13.0000 - val_accuracy: 0.9812 - val_precision: 0.0706 - val_recall: 0.8312 - val_auc: 0.9712
Epoch 53/1000
20/20 [==============================] - 0s 23ms/step - loss: 0.1316 - tp: 19347.0000 - fp: 657.0000 - tn: 19790.0000 - fn: 1166.0000 - accuracy: 0.9555 - precision: 0.9672 - recall: 0.9432 - auc: 0.9893 - val_loss: 0.0785 - val_tp: 64.0000 - val_fp: 810.0000 - val_tn: 44682.0000 - val_fn: 13.0000 - val_accuracy: 0.9819 - val_precision: 0.0732 - val_recall: 0.8312 - val_auc: 0.9713
Epoch 54/1000
20/20 [==============================] - 1s 30ms/step - loss: 0.1288 - tp: 19388.0000 - fp: 632.0000 - tn: 19757.0000 - fn: 1183.0000 - accuracy: 0.9557 - precision: 0.9684 - recall: 0.9425 - auc: 0.9898 - val_loss: 0.0774 - val_tp: 64.0000 - val_fp: 806.0000 - val_tn: 44686.0000 - val_fn: 13.0000 - val_accuracy: 0.9820 - val_precision: 0.0736 - val_recall: 0.8312 - val_auc: 0.9718
Epoch 55/1000
20/20 [==============================] - 1s 26ms/step - loss: 0.1279 - tp: 19208.0000 - fp: 644.0000 - tn: 19941.0000 - fn: 1167.0000 - accuracy: 0.9558 - precision: 0.9676 - recall: 0.9427 - auc: 0.9898 - val_loss: 0.0768 - val_tp: 65.0000 - val_fp: 804.0000 - val_tn: 44688.0000 - val_fn: 12.0000 - val_accuracy: 0.9821 - val_precision: 0.0748 - val_recall: 0.8442 - val_auc: 0.9718
Epoch 56/1000
20/20 [==============================] - 1s 26ms/step - loss: 0.1274 - tp: 19518.0000 - fp: 630.0000 - tn: 19651.0000 - fn: 1161.0000 - accuracy: 0.9563 - precision: 0.9687 - recall: 0.9439 - auc: 0.9901 - val_loss: 0.0759 - val_tp: 65.0000 - val_fp: 794.0000 - val_tn: 44698.0000 - val_fn: 12.0000 - val_accuracy: 0.9823 - val_precision: 0.0757 - val_recall: 0.8442 - val_auc: 0.9720
Epoch 57/1000
20/20 [==============================] - 1s 27ms/step - loss: 0.1274 - tp: 19309.0000 - fp: 647.0000 - tn: 19867.0000 - fn: 1137.0000 - accuracy: 0.9564 - precision: 0.9676 - recall: 0.9444 - auc: 0.9899 - val_loss: 0.0749 - val_tp: 65.0000 - val_fp: 780.0000 - val_tn: 44712.0000 - val_fn: 12.0000 - val_accuracy: 0.9826 - val_precision: 0.0769 - val_recall: 0.8442 - val_auc: 0.9717
Epoch 58/1000
20/20 [==============================] - 0s 23ms/step - loss: 0.1237 - tp: 19236.0000 - fp: 576.0000 - tn: 20026.0000 - fn: 1122.0000 - accuracy: 0.9585 - precision: 0.9709 - recall: 0.9449 - auc: 0.9905 - val_loss: 0.0747 - val_tp: 65.0000 - val_fp: 794.0000 - val_tn: 44698.0000 - val_fn: 12.0000 - val_accuracy: 0.9823 - val_precision: 0.0757 - val_recall: 0.8442 - val_auc: 0.9710
Epoch 59/1000
20/20 [==============================] - 0s 23ms/step - loss: 0.1240 - tp: 19363.0000 - fp: 602.0000 - tn: 19855.0000 - fn: 1140.0000 - accuracy: 0.9575 - precision: 0.9698 - recall: 0.9444 - auc: 0.9906 - val_loss: 0.0743 - val_tp: 65.0000 - val_fp: 793.0000 - val_tn: 44699.0000 - val_fn: 12.0000 - val_accuracy: 0.9823 - val_precision: 0.0758 - val_recall: 0.8442 - val_auc: 0.9715
Epoch 60/1000
19/20 [===========================>..] - ETA: 0s - loss: 0.1202 - tp: 18282.0000 - fp: 589.0000 - tn: 18983.0000 - fn: 1058.0000 - accuracy: 0.9577 - precision: 0.9688 - recall: 0.9453 - auc: 0.9911Restoring model weights from the end of the best epoch.
20/20 [==============================] - 0s 22ms/step - loss: 0.1205 - tp: 19230.0000 - fp: 629.0000 - tn: 19987.0000 - fn: 1114.0000 - accuracy: 0.9574 - precision: 0.9683 - recall: 0.9452 - auc: 0.9910 - val_loss: 0.0734 - val_tp: 65.0000 - val_fp: 791.0000 - val_tn: 44701.0000 - val_fn: 12.0000 - val_accuracy: 0.9824 - val_precision: 0.0759 - val_recall: 0.8442 - val_auc: 0.9719
Epoch 00060: early stopping

Re-check training history

plot_metrics(resampled_history)

png

Evaluate metrics

train_predictions_resampled = resampled_model.predict(train_features, batch_size=BATCH_SIZE)
test_predictions_resampled = resampled_model.predict(test_features, batch_size=BATCH_SIZE)

resampled_results = resampled_model.evaluate(test_features, test_labels,
                                             batch_size=BATCH_SIZE, verbose=0)
for name, value in zip(resampled_model.metrics_names, resampled_results):
  print(name, ': ', value)
print()

plot_cm(test_labels, test_predictions_weighted)

loss :  0.08324556274180048
tp :  89.0
fp :  1138.0
tn :  55726.0
fn :  9.0
accuracy :  0.97986376
precision :  0.072534636
recall :  0.90816325
auc :  0.9878597

Legitimate Transactions Detected (True Negatives):  55921
Legitimate Transactions Incorrectly Detected (False Positives):  943
Fraudulent Transactions Missed (False Negatives):  10
Fraudulent Transactions Detected (True Positives):  88
Total Fraudulent Transactions:  98

png

Plot the ROC

plot_roc("Train Baseline", train_labels, train_predictions_baseline, color=colors[0])
plot_roc("Test Baseline", test_labels, test_predictions_baseline, color=colors[0], linestyle='--')

plot_roc("Train Weighted", train_labels, train_predictions_weighted, color=colors[1])
plot_roc("Test Weighted", test_labels, test_predictions_weighted, color=colors[1], linestyle='--')

plot_roc("Train Resampled", train_labels, train_predictions_resampled,  color=colors[2])
plot_roc("Test Resampled", test_labels, test_predictions_resampled,  color=colors[2], linestyle='--')
plt.legend(loc='lower right')

<matplotlib.legend.Legend at 0x7fee383c5ba8>

png

Applying this tutorial to your problem

Imbalanced data classification is an inherantly difficult task since there are so few samples to learn from. You should always start with the data first and do your best to collect as many samples as possible and give substantial thought to what features may be relevant so the model can get the most out of your minority class. At some point your model may struggle to improve and yield the results you want, so it is important to keep in mind the context of your problem and the trade offs between different types of errors.