Updating the installation steps.

PiperOrigin-RevId: 354248645

ledge/README.md

@@ -7,25 +7,32 @@ thirty-fifth Conference on Artificial Intelligence (AAAI-21).
 
 This code implements the Function Approximator example from the paper,
 approximately reconstructing a 1-D function. Following the paper, the
-results of step 1 and step 2 are provided in the ckpts directory. This
+results of step 1 and step 2 are provided in the data package. This
 code performs:
 
-  Step 3: Correct the residual error.
-  Step 4: Extract the interpretable commands. Commands are printed
-    to the console.
-  Step 5: Fine-tune the extracted commands.
-
-To install, untar the data package in the same directory as ledge.py:
+- Step 3: Correct the residual error.
+- Step 4: Extract the interpretable commands. Commands are printed to the console.
+- Step 5: Fine-tune the extracted commands.
 
+## Install the Code
+From your working directory, run:
 ```
+svn export https://github.com/google-research/google-research/trunk/ledge
+cd ledge
 pip3 install -r requirements.txt
-tar zxf data.tar.gz
 ```
 
-As a simple demo:
+## Install the Data
+```
+wget http://storage.googleapis.com/gresearch/ledge/data2.tar.gz
+tar xzf data2.tar.gz
+rm data2.tar.gz
+```
+
+## Run a Simple Demo
 ```
 python -m ledge --target_function_file targets/approximate_20
-eog /tmp/refit_fig0.png
+eog /tmp/refit.png
 ```
 
 For a list of flags and options:

ledge/requirements.txt

@@ -1,41 +1,7 @@
 absl-py==0.11.0
-astunparse==1.6.3
-cachetools==4.2.0
-certifi==2020.12.5
-chardet==4.0.0
-cycler==0.10.0
-flatbuffers==1.12
-gast==0.3.3
-google-auth==1.24.0
-google-auth-oauthlib==0.4.2
-google-pasta==0.2.0
-grpcio==1.32.0
-h5py==2.10.0
-idna==2.10
 Keras-Preprocessing==1.1.2
-kiwisolver==1.3.1
-Markdown==3.3.3
 matplotlib==3.3.3
 numpy==1.19.4
-oauthlib==3.1.0
-opt-einsum==3.3.0
 Pillow==8.0.1
-protobuf==3.14.0
-pyasn1==0.4.8
-pyasn1-modules==0.2.8
-pyparsing==2.4.7
-python-dateutil==2.8.1
-requests==2.25.1
-requests-oauthlib==1.3.0
-rsa==4.6
-six==1.15.0
-tensorboard==2.4.0
-tensorboard-plugin-wit==1.7.0
 tensorflow==2.4.0
-tensorflow-estimator==2.4.0
-termcolor==1.1.0
 tf-slim==1.1.0
-typing-extensions==3.7.4.3
-urllib3==1.26.2
-Werkzeug==1.0.1
-wrapt==1.12.1