Skip to content

Commit

Permalink
Apply changes to example that follow from change to the dataset.
Browse files Browse the repository at this point in the history
Age is now also a feature, but we're not using it here.
  • Loading branch information
arokem committed Dec 7, 2024
1 parent 2178c22 commit 9b98579
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion examples/plot_als_classification.py
Original file line number Diff line number Diff line change
Expand Up @@ -54,6 +54,7 @@

X = afqdata.X
y = afqdata.y.astype(float) # SGL expects float targets
is_als = y[:, 0]
groups = afqdata.groups
feature_names = afqdata.feature_names
group_names = afqdata.group_names
Expand Down Expand Up @@ -117,7 +118,7 @@
# scikit-learn functions

scores = cross_validate(
pipe, X, y, cv=5, return_train_score=True, return_estimator=True
pipe, X, is_als, cv=5, return_train_score=True, return_estimator=True
)

# Display results
Expand Down

0 comments on commit 9b98579

Please sign in to comment.