Project import generated by Copybara.

PiperOrigin-RevId: 355228987
KarthikDutt · Feb 2, 2021 · 665501e · 665501e
1 parent ac05b95
commit 665501e
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/README.md b/README.md
@@ -32,6 +32,7 @@ Below is a code snippet for doing so:
 
 ```python
 import model_search
+from model_search import constants
 from model_search import single_trainer
 from model_search.data import csv_data
 
@@ -41,7 +42,7 @@ trainer = single_trainer.SingleTrainer(
         logits_dimension=2,
         record_defaults=[0, 0, 0, 0],
         filename="model_search/data/testdata/csv_random_data.csv")),
-    spec=model_search.constants.DEFAULT_DNN)
+    spec=constants.DEFAULT_DNN)
 
 trainer.try_models(
     number_models=200,

diff --git a/requirements.txt b/requirements.txt
@@ -0,0 +1,8 @@
+six==1.15.0
+sklearn==0.0
+tensorflow==2.2.0
+absl-py==0.10.0
+tf-slim==1.1.0
+ml-metadata==0.26.0
+keras-tuner==1.0.2
+mock==4.0.3