From 916394b85cbb598b179aa47e623ee2bd659d3061 Mon Sep 17 00:00:00 2001 From: harrisonfloam Date: Thu, 26 Sep 2024 21:06:43 -0400 Subject: [PATCH] add FourPeaksGenerator class imports --- mlrose_hiive/__init__.py | 2 +- mlrose_hiive/generators/__init__.py | 1 + 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/mlrose_hiive/__init__.py b/mlrose_hiive/__init__.py index 5c462e1b..0526ffbf 100644 --- a/mlrose_hiive/__init__.py +++ b/mlrose_hiive/__init__.py @@ -27,7 +27,7 @@ from .runners import GARunner, MIMICRunner, RHCRunner, SARunner, NNGSRunner, SKMLPRunner from .runners import (build_data_filename) from .generators import (MaxKColorGenerator, QueensGenerator, FlipFlopGenerator, TSPGenerator, KnapsackGenerator, - ContinuousPeaksGenerator) + ContinuousPeaksGenerator, FourPeaksGenerator) from .samples import SyntheticData from .samples import (plot_synthetic_dataset) diff --git a/mlrose_hiive/generators/__init__.py b/mlrose_hiive/generators/__init__.py index ea7eb10c..31a84bb6 100644 --- a/mlrose_hiive/generators/__init__.py +++ b/mlrose_hiive/generators/__init__.py @@ -9,3 +9,4 @@ from .queens_generator import QueensGenerator from .tsp_generator import TSPGenerator from .continuous_peaks_generator import ContinuousPeaksGenerator +from .four_peaks_generator import FourPeaksGenerator