Skip to content
Snippets Groups Projects
Commit 441d4d17 authored by Tizian Wenzel's avatar Tizian Wenzel
Browse files

v0.1.1: Minor updates, reduced printing.

parent b540ded6
Branches
Tags
No related merge requests found
...@@ -5,7 +5,7 @@ Python implementation of the 2L-VKOGA algorithm, which uses a kernel optimizatio ...@@ -5,7 +5,7 @@ Python implementation of the 2L-VKOGA algorithm, which uses a kernel optimizatio
## Installation ## Installation
pip install git+https://gitlab.mathematik.uni-stuttgart.de/pub/ians-anm/2l-vkoga@v0.1.0 pip install git+https://gitlab.mathematik.uni-stuttgart.de/pub/ians-anm/2l-vkoga@v0.1.1
## Usage ## Usage
......
...@@ -8,7 +8,7 @@ setup( ...@@ -8,7 +8,7 @@ setup(
author='Tizian Wenzel', author='Tizian Wenzel',
author_email='tizian.wenzel@mathematik.uni-stuttgart.de', author_email='tizian.wenzel@mathematik.uni-stuttgart.de',
license='GNU v3.ß', license='GNU v3.ß',
packages=['vkoga_2l'], packages=['vkoga_2L'],
install_requires=['numpy==1.24.1', 'torch==2.1.1', 'matplotlib==3.7.2', 'scipy==1.11.4', 'scikit-learn==1.3.2'], install_requires=['numpy==1.24.1', 'torch==2.1.1', 'matplotlib==3.7.2', 'scipy==1.11.4', 'scikit-learn==1.3.2'],
classifiers=[ classifiers=[
'Development Status :: 1 - Planning', 'Development Status :: 1 - Planning',
......
File moved
File moved
File moved
...@@ -142,7 +142,7 @@ class VKOGA_2L(BaseEstimator): ...@@ -142,7 +142,7 @@ class VKOGA_2L(BaseEstimator):
n_folds=None, flag_initialize_diagonal=True, n_folds=None, flag_initialize_diagonal=True,
flag_symmetric_A=False) flag_symmetric_A=False)
model_OptimKernel.optimize(torch.from_numpy(X).float(), torch.from_numpy(y).float(), model_OptimKernel.optimize(torch.from_numpy(X).float(), torch.from_numpy(y).float(),
flag_optim_verbose=True) flag_optim_verbose=self.verbose)
self.A = model_OptimKernel.A.detach().numpy() self.A = model_OptimKernel.A.detach().numpy()
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment