Update README.md

imatge-upc · Jan 17, 2023 · 1aefbb8 · 1aefbb8
1 parent 5565399
commit 1aefbb8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -1,7 +1,7 @@
 # SurvLIMEpy
 
 <p align="center">
-    <img src="logo.png" width="256" height="256">
+    <img src="https://github.com/imatge-upc/SurvLIMEpy/blob/main/logo.png?raw=true" width="256" height="256">
 </p>
 
 
@@ -27,7 +27,7 @@ loader = Loader(dataset_name='udca')
 X, events, times = loader.load_data()
 
 # Train a model
-train, val, test = loader.preprocess_datasets(X, events, times)
+train, test = loader.preprocess_datasets(X, events, times)
 model = CoxPHSurvivalAnalysis()
 model.fit(train[0], train[1])