Skip to content

Commit 0abcd79

Browse files
Updated polynomial evaluation.
1 parent a1c107d commit 0abcd79

1 file changed

Lines changed: 5 additions & 4 deletions

File tree

scripts/run_polynomial_evaluation.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -6,12 +6,12 @@
66
import os
77
import numpy as np # type: ignore
88

9-
NUM_EPOCH = 20000
9+
NUM_EPOCH = 2000
1010
TARGET_COLUMN = "target" # Assuming the target column is named 'target'
1111
NUM_DEPENDENT_FEATURE = 6
1212
NUM_OUTPUT_FEATURE = 2
1313
NUM_INPUT_FEATURE = NUM_DEPENDENT_FEATURE + NUM_OUTPUT_FEATURE
14-
NUM_SAMPLE = 1000
14+
NUM_SAMPLE = 10000
1515

1616
def makeModel():
1717
"""Creates a model for testing."""
@@ -20,7 +20,7 @@ def makeModel():
2020

2121
TRAIN_DL, dct = makeAutocoderData(num_sample=NUM_SAMPLE, num_independent_feature=NUM_OUTPUT_FEATURE,
2222
num_dependent_feature=NUM_DEPENDENT_FEATURE, num_value=10)
23-
TEST_DL = makeAutocoderData(num_sample=1000, num_independent_feature=NUM_OUTPUT_FEATURE,
23+
TEST_DL = makeAutocoderData(num_sample=10000, num_independent_feature=NUM_OUTPUT_FEATURE,
2424
num_dependent_feature=NUM_DEPENDENT_FEATURE, num_value=10,
2525
multiplier_dct=dct)[0]
2626
SCRIPT_DIR = os.path.abspath(os.path.dirname(__file__))
@@ -77,4 +77,5 @@ def compareModels(num_models: int = 10):
7777
if __name__ == "__main__":
7878
#train()
7979
#evaluate()
80-
compareModels()
80+
makeModels()
81+
#compareModels()

0 commit comments

Comments
 (0)