Browse Source

disable cross validation check

tags/v0.3.1
Frozenmad 5 years ago
parent
commit
b9efe52562
1 changed files with 2 additions and 2 deletions
  1. +2
    -2
      autogl/solver/classifier/graph_classifier.py

+ 2
- 2
autogl/solver/classifier/graph_classifier.py View File

@@ -266,8 +266,8 @@ class AutoGraphClassifier(BaseClassifier):
"Please manually pass train and val ratio."
)
LOGGER.info("Use the default train/val/test ratio in given dataset")
if hasattr(dataset.train_split, "n_splits"):
cross_validation = True
#if hasattr(dataset.train_split, "n_splits"):
# cross_validation = True

elif train_split is not None and val_split is not None:
utils.graph_random_splits(dataset, train_split, val_split, seed=seed)


Loading…
Cancel
Save