From b3d16d72f58b40c34b79d082f305658e0c242940 Mon Sep 17 00:00:00 2001
From: Tizian Wenzel <wenzeltn@nbanm02.mathematik.uni-stuttgart.de>
Date: Thu, 8 Jun 2023 13:20:15 +0200
Subject: [PATCH] Some more updates.

---
 section_4.1_compute_visualize.py | 5 ++++-
 setup.sh                         | 4 ++--
 2 files changed, 6 insertions(+), 3 deletions(-)

diff --git a/section_4.1_compute_visualize.py b/section_4.1_compute_visualize.py
index 3f7e0e7..56acdfe 100644
--- a/section_4.1_compute_visualize.py
+++ b/section_4.1_compute_visualize.py
@@ -12,6 +12,7 @@ import numpy as np
 from matplotlib import pyplot as plt
 from scipy import io
 import os
+import matplotlib
 
 
 np.random.seed(1)
@@ -70,6 +71,8 @@ io.savemat(path_for_results + name_dataset + '.mat',
 
 
 ## Some plots
+matplotlib.use('TKAgg')
+
 R = np.linspace(0, 1, int(1.5*array_cv_f.shape[1]))
 array_color=plt.cm.hsv(R)
 
@@ -103,7 +106,7 @@ plt.legend(list_legend + ['A[0,1]', 'A[1,0]'])
 plt.title('matrix entries')
 plt.xscale('log')
 plt.xlabel('iterations')
-plt.draw()
+plt.show()
 
 
 
diff --git a/setup.sh b/setup.sh
index a454aeb..ddd3cd6 100755
--- a/setup.sh
+++ b/setup.sh
@@ -16,8 +16,8 @@ source venv/bin/activate
 pip install --upgrade pip
 pip3 install --ignore-installed torch==1.9.1+cpu torchvision==0.10.1+cpu torchaudio==0.9.1 -f https://download.pytorch.org/whl/torch_stable.html
 
-scipy==1.7.3
-sklearn==0.0.post5
+pip3 install scipy==1.7.3
+pip3 install scikit-learn==1.0.2
 
 
 # Install further stuff
-- 
GitLab