Some more updates.

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()
 
 
 

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