import matplotlib.pyplot as plt

# Monitor training progress
train_losses = []
val_losses = []

for epoch in range(max_epochs):
    # Training step
    clf.partial_fit(X_train_batch, y_train_batch)

    # Record losses
    train_loss = clf.score(X_train_batch, y_train_batch)
    val_loss = clf.score(X_val, y_val)

    train_losses.append(train_loss)
    val_losses.append(val_loss)

# Plot convergence
plt.plot(train_losses, label='Training')
plt.plot(val_losses, label='Validation')
plt.legend()
plt.show()

        __
        __