From 2619f041d771a479a8a70a788bad3e92cbe098b6 Mon Sep 17 00:00:00 2001 From: Krzysztof Rudnicki Date: Mon, 25 Nov 2024 15:47:17 +0100 Subject: [PATCH] feat: calculate eigenvalues using numpy library --- code/richardson_method.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/code/richardson_method.py b/code/richardson_method.py index 34e836ea..089ac8cd 100644 --- a/code/richardson_method.py +++ b/code/richardson_method.py @@ -5,6 +5,7 @@ from processing_type import ProcessingType from time_measurement import time_measurement, time_accumulator, tests_time import time import gc +import numpy as np class RichardsonMethod: @time_measurement(time_accumulator) @@ -23,7 +24,10 @@ class RichardsonMethod: @staticmethod def calculate_eigenvalues(LinAlgType, A, type): - return EigenvalueMethods.inverse_power_method(LinAlgType, A, type), EigenvalueMethods.power_method(LinAlgType, A, type) + eigenvalues = np.linalg.eigvals(A) + lambda_min = np.min(eigenvalues) + lambda_max = np.max(eigenvalues) + return lambda_min, lambda_max @staticmethod def calculate_omega(lambda_min, lambda_max):