Reduce isolated folders in model garden.

Move movielens to recommendation

PiperOrigin-RevId: 269680664

official/r1/wide_deep/movielens_dataset.py

@@ -28,7 +28,7 @@ import numpy as np
 import tensorflow as tf
 # pylint: enable=wrong-import-order
 
-from official.datasets import movielens
+from official.recommendation import movielens
 from official.r1.utils.data import file_io
 from official.utils.flags import core as flags_core
 

official/r1/wide_deep/movielens_main.py

@@ -24,7 +24,7 @@ from absl import app as absl_app
 from absl import flags
 import tensorflow as tf
 
-from official.datasets import movielens
+from official.recommendation import movielens
 from official.utils.flags import core as flags_core
 from official.utils.logs import logger
 from official.r1.wide_deep import movielens_dataset

official/r1/wide_deep/movielens_test.py

@@ -23,7 +23,7 @@ import unittest
 import numpy as np
 import tensorflow as tf  # pylint: disable=g-bad-import-order
 
-from official.datasets import movielens
+from official.recommendation import movielens
 from official.utils.misc import keras_utils
 from official.utils.testing import integration
 from official.r1.wide_deep import movielens_dataset

official/recommendation/constants.py

@@ -18,7 +18,7 @@ import sys
 
 import numpy as np
 
-from official.datasets import movielens
+from official.recommendation import movielens
 
 # ==============================================================================
 # == Main Thread Data Processing ===============================================

official/recommendation/create_ncf_data.py

@@ -26,7 +26,7 @@ from absl import flags
 import tensorflow.compat.v2 as tf
 # pylint: enable=g-bad-import-order
 
-from official.datasets import movielens
+from official.recommendation import movielens
 from official.recommendation import data_preprocessing
 
 flags.DEFINE_string(

official/recommendation/data_pipeline.py

@@ -35,8 +35,8 @@ from six.moves import queue
 import tensorflow as tf
 from absl import logging
 
-from official.datasets import movielens
 from official.recommendation import constants as rconst
+from official.recommendation import movielens
 from official.recommendation import popen_helper
 from official.recommendation import stat_utils
 

official/recommendation/data_preprocessing.py

@@ -31,9 +31,9 @@ import tensorflow as tf
 from absl import logging
 # pylint: enable=wrong-import-order
 
-from official.datasets import movielens
 from official.recommendation import constants as rconst
 from official.recommendation import data_pipeline
+from official.recommendation import movielens
 from official.utils.logs import mlperf_helper
 
 

official/recommendation/data_test.py

@@ -27,9 +27,9 @@ import numpy as np
 import scipy.stats
 import tensorflow as tf
 
-from official.datasets import movielens
 from official.recommendation import constants as rconst
 from official.recommendation import data_preprocessing
+from official.recommendation import movielens
 from official.recommendation import popen_helper
 from official.utils.misc import keras_utils
 

official/recommendation/ncf_common.py

@@ -29,10 +29,10 @@ from absl import logging
 import tensorflow as tf
 # pylint: enable=g-bad-import-order
 
-from official.datasets import movielens
 from official.recommendation import constants as rconst
 from official.recommendation import data_pipeline
 from official.recommendation import data_preprocessing
+from official.recommendation import movielens
 from official.utils.flags import core as flags_core
 from official.utils.misc import distribution_utils
 from official.utils.misc import keras_utils

official/recommendation/ncf_estimator_main.py

@@ -39,10 +39,10 @@ from absl import logging
 import tensorflow as tf
 # pylint: enable=g-bad-import-order
 
-from official.datasets import movielens
 from official.recommendation import constants as rconst
 from official.recommendation import data_pipeline
 from official.recommendation import data_preprocessing
+from official.recommendation import movielens
 from official.recommendation import ncf_common
 from official.recommendation import neumf_model
 from official.utils.flags import core as flags_core

official/recommendation/ncf_input_pipeline.py

@@ -24,8 +24,8 @@ import functools
 import tensorflow.compat.v2 as tf
 # pylint: enable=g-bad-import-order
 
-from official.datasets import movielens
 from official.recommendation import constants as rconst
+from official.recommendation import movielens
 from official.recommendation import data_pipeline
 
 NUM_SHARDS = 16

official/recommendation/ncf_keras_main.py

@@ -32,8 +32,8 @@ from absl import logging
 import tensorflow as tf
 # pylint: enable=g-bad-import-order
 
-from official.datasets import movielens
 from official.recommendation import constants as rconst
+from official.recommendation import movielens
 from official.recommendation import ncf_common
 from official.recommendation import ncf_input_pipeline
 from official.recommendation import neumf_model

official/recommendation/neumf_model.py

@@ -38,8 +38,8 @@ import sys
 from six.moves import xrange  # pylint: disable=redefined-builtin
 import tensorflow as tf
 
-from official.datasets import movielens  # pylint: disable=g-bad-import-order
 from official.recommendation import constants as rconst
+from official.recommendation import movielens
 from official.recommendation import ncf_common
 from official.recommendation import stat_utils
 from official.utils.logs import mlperf_helper