diff --git a/EIVPackage/EIVData/linear.py b/EIVPackage/EIVData/linear.py index 5dc363baa5a62d0a5a8858c2e5d169eb5b8e8d55..b1825905acafc69332d1c01418eb7db00e0f28eb 100644 --- a/EIVPackage/EIVData/linear.py +++ b/EIVPackage/EIVData/linear.py @@ -2,11 +2,11 @@ import torch import sys from torch.utils.data import TensorDataset, random_split -total_number_of_datapoints = 10000 +total_number_of_datapoints = 2000 input_range = [-1,1] slope = 1.0 intercept = 0.0 -x_noise_strength = 0.1 +x_noise_strength = 0.05 y_noise_strength = 0.1 def get_normalization(*args): diff --git a/Experiments/configurations/eiv_linear.json b/Experiments/configurations/eiv_linear.json index 92c292ef49d22cfbd7ffc561e6f64c015f3e35f3..a83fc15d5a0d1ba30017465b984ab6b79035b60b 100644 --- a/Experiments/configurations/eiv_linear.json +++ b/Experiments/configurations/eiv_linear.json @@ -2,7 +2,7 @@ "long_dataname": "linear", "short_dataname": "linear", "lr": 1e-3, - "batch_size": 100, + "batch_size": 64, "test_batch_size": 800, "number_of_epochs": 100, "unscaled_reg": 10, @@ -14,8 +14,8 @@ "eiv_prediction_number_of_batches": 10, "init_std_y_list": [0.5], "gamma": 0.5, - "hidden_layers": [1024, 1024, 1024, 1024], - "fixed_std_x": 0.1, + "hidden_layers": [128, 128, 128, 128], + "fixed_std_x": 0.05, "seed_range": [0,10], "gpu_number": 1 } diff --git a/Experiments/configurations/noneiv_linear.json b/Experiments/configurations/noneiv_linear.json index 67c6ea438f7a6fbead0e0a350e69560fd86fa1ff..1b2110a6149f8125fba9752a95cc134fac7267c2 100644 --- a/Experiments/configurations/noneiv_linear.json +++ b/Experiments/configurations/noneiv_linear.json @@ -2,7 +2,7 @@ "long_dataname": "linear", "short_dataname": "linear", "lr": 1e-3, - "batch_size": 100, + "batch_size": 64, "test_batch_size": 800, "number_of_epochs": 100, "unscaled_reg": 10, @@ -14,7 +14,7 @@ "noneiv_prediction_number_of_batches": 10, "init_std_y_list": [0.5], "gamma": 0.5, - "hidden_layers": [1024, 1024, 1024, 1024], + "hidden_layers": [128, 128, 128, 128], "seed_range": [0,10], "gpu_number": 1 }