Skip to content
Snippets Groups Projects
Commit 6d39cf45 authored by v.gramlich1's avatar v.gramlich1
Browse files

Add imports to run_with_finetuning.py...

Add imports to run_with_finetuning.py IntelliO3_ts_architecture_finetune_all_dense, IntelliO3_ts_architecture_finetune_outputs and IntelliO3_ts_architecture_finetune_main_output
parent ad811eef
No related branches found
No related tags found
1 merge request!302Draft: Resolve "Class-based Oversampling technique"
Pipeline #77541 passed
......@@ -5,7 +5,8 @@ import argparse
from mlair.workflows import DefaultWorkflow
from mlair.helpers import remove_items
from mlair.configuration.defaults import DEFAULT_PLOT_LIST
from mlair.model_modules.model_class import IntelliO3_ts_architecture, IntelliO3_ts_architecture_freeze
from mlair.model_modules.model_class import IntelliO3_ts_architecture, IntelliO3_ts_architecture_finetune_all_dense, \
IntelliO3_ts_architecture_finetune_outputs, IntelliO3_ts_architecture_finetune_main_output
import os
......@@ -27,7 +28,7 @@ def main(parser_args):
#stations=["DEBW013", "DEBW087"],
epochs=1, external_weights="/home/vincentgramlich/mlair/data/weights/testrun_network_daily_model-best.h5",
train_model=True, create_new_model=True, network="UBA",
model=IntelliO3_ts_architecture_freeze,
model=IntelliO3_ts_architecture_finetune_all_dense,
window_lead_time=1,
#oversampling_method="bin_oversampling", oversampling_bins=10, oversampling_rates_cap=100, window_lead_time=2,
evaluate_bootstraps=False, plot_list=["PlotContingency"],
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment