steffen/hom_traj_gen
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Refactoring ML_Lib

Browse Source
This commit is contained in:
Si11ium
2020-04-08 14:50:16 +02:00
parent 25c0e8e358
commit 27660c1458
32 changed files with 70 additions and 79 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
models/homotopy_classification/cnn_based.py
View File

@ -8,9 +8,9 @@ from torch.optim import Adam
from torch.utils.data import DataLoader
from datasets.trajectory_dataset import TrajData
from lib.evaluation.classification import ROCEvaluation
from lib.modules.utils import LightningBaseModule, Flatten
from lib.modules.blocks import ConvModule, ResidualModule
from ml_lib.evaluation.classification import ROCEvaluation
from ml_lib.modules.utils import LightningBaseModule, Flatten
from ml_lib.modules.blocks import ConvModule, ResidualModule
import matplotlib.pyplot as plt