updated readme (#208)

* updated_readme * updated_readme * updated_readme * updated_readme * updated_readme * updated_readme * updated_readme Co-authored-by: Jonthan Liberman <jonthanliberman@ip-192-168-1-20.eu-west-1.compute.internal> Co-authored-by: Jonthan Liberman <jonthanliberman@ip-192-168-1-1.eu-west-1.compute.internal>
deepchecks · Nov 30, 2021 · 8f4b078 · 8f4b078
1 parent f1f9769
commit 8f4b078
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/README.md b/README.md
@@ -56,7 +56,7 @@ First clone the repository and then install the package from inside the reposito
 git clone https://github.com/deepchecks/deepchecks.git
 cd deepchecks
 # for installing stable tag version and not the latest commit to main
-git checkout tags/<name_of_latest_tag>
+git checkout tags/<version>
 ```
 and then either:
 ```bash
@@ -136,12 +136,12 @@ Let's take the "iris" dataset as an example:
 ```python
 import pandas as pd
 from sklearn.datasets import load_iris
+from sklearn.model_selection import train_test_split
 ```
 ```python
 iris_df = load_iris(return_X_y=False, as_frame=True)['frame']
-train_len = round(0.67*len(iris_df))
-df_train = iris_df[:train_len]
-df_test = iris_df[train_len:]
+label_col = 'target'
+df_train, df_test = train_test_split(iris_df, stratify=iris_df[label_col], random_state=0)
 ```
 To run an existing suite all you need to do is import the suite and run it -