Skip to content

Commit

Permalink
README
Browse files Browse the repository at this point in the history
  • Loading branch information
amaiya committed Jun 19, 2024
1 parent 238d353 commit b6fcf24
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -316,8 +316,8 @@ learner.evaluate(tst, class_names=preproc.get_classes())
3. Install *ktrain*: `pip install ktrain`

4. If using `tensorflow>=2.16`:
- install **tf_keras** (`pip install tf_keras`)
- set the environment variable `TF_USE_LEGACY_KERAS` to true before importing **ktrain** (e.g., add `export TF_USE_LEGACY_KERAS=1` in `.bashrc` for Linux or put `os.environ["TF_USE_LEGACY_KERAS"]="1"` at top of your code).
a. install **tf_keras**: `pip install tf_keras`
b. set the environment variable `TF_USE_LEGACY_KERAS` to true before importing **ktrain**


The above should be all you need on Linux systems and cloud computing environments like Google Colab and AWS EC2. If you are using **ktrain** on a **Windows computer**, you can follow these
Expand Down

0 comments on commit b6fcf24

Please sign in to comment.