Skip to content

Commit

Permalink
Merge pull request #1119 from mollerhoj/patch-1
Browse files Browse the repository at this point in the history
Update evaluate.py
  • Loading branch information
tmbo committed Jun 11, 2018
2 parents 52ffa64 + a2e9c3e commit 0c0984a
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion rasa_nlu/evaluate.py
Original file line number Diff line number Diff line change
Expand Up @@ -533,7 +533,7 @@ def generate_folds(n, td):
"""Generates n cross validation folds for training data td."""

from sklearn.model_selection import StratifiedKFold
skf = StratifiedKFold(n_splits=n, random_state=2018, shuffle=True)
skf = StratifiedKFold(n_splits=n, shuffle=True)
x = td.intent_examples
y = [example.get("intent") for example in x]
for i_fold, (train_index, test_index) in enumerate(skf.split(x, y)):
Expand Down

0 comments on commit 0c0984a

Please sign in to comment.