Machine Learning in Elixir: Missing option in confusion matrix (page 104)

Pragmatic Bookshelf PragProg Customers
,
1

Version: B1.0

On page 104 the code:

Scholar.Metrics.confusion_matrix(test_targets, test_preds)

throws an error: required :num_classes option not found, received options: []

I think the code should be:

Scholar.Metrics.confusion_matrix(test_targets, test_preds, num_classes: 3)

Another similar error on page 107, the following code fails:

model = Scholar.Neighbors.KNearestNeighbors.fit(train_inputs, train_targets)

I think it should be:

model = Scholar.Neighbors.KNearestNeighbors.fit(train_inputs, train_targets, num_classes: 3)