VictorMylle/Thesis
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Changed imports

Browse Source
This commit is contained in:
Victor Mylle
2023-11-26 00:29:35 +00:00
parent 360e9f4e8e
commit f0c6369dd3
7 changed files with 35 additions and 41 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
src/notebooks/training.py
View File

@@ -1,13 +1,11 @@
import sys
sys.path.append('..')
from data import DataProcessor, DataConfig
from trainers.quantile_trainer import AutoRegressiveQuantileTrainer, NonAutoRegressiveQuantileRegression
from trainers.probabilistic_baseline import ProbabilisticBaselineTrainer
from trainers.autoregressive_trainer import AutoRegressiveTrainer
from trainers.trainer import Trainer
from utils.clearml import ClearMLHelper
from models import *
from losses import *
from src.data import DataProcessor, DataConfig
from src.trainers.quantile_trainer import AutoRegressiveQuantileTrainer, NonAutoRegressiveQuantileRegression
from src.trainers.probabilistic_baseline import ProbabilisticBaselineTrainer
from src.trainers.autoregressive_trainer import AutoRegressiveTrainer
from src.trainers.trainer import Trainer
from src.utils.clearml import ClearMLHelper
from src.models import *
from src.losses import *
import torch
import numpy as np
from torch.nn import MSELoss, L1Loss
@@ -15,10 +13,6 @@ from datetime import datetime
import pytz
import torch.nn as nn
# auto reload
%load_ext autoreload
%autoreload 2
#### ClearML ####
clearml_helper = ClearMLHelper(project_name="Thesis/NrvForecast")