Commit
·
4f3b892
1
Parent(s):
0b58194
updates
Browse files
script.py
CHANGED
@@ -115,13 +115,6 @@ def train_model(training_config=training_config):
|
|
115 |
NOTE: DO NOT MODIFY THIS FUNCTION.
|
116 |
"""
|
117 |
|
118 |
-
script_dir = os.path.dirname(os.path.abspath(__file__))
|
119 |
-
|
120 |
-
config_path = os.path.join(script_dir, 'training_config.yaml')
|
121 |
-
|
122 |
-
with open(config_path, 'r') as file:
|
123 |
-
training_config = yaml.safe_load(file)
|
124 |
-
|
125 |
training_dataset = prepare_dataset()
|
126 |
|
127 |
print("Splitting the dataset...")
|
|
|
115 |
NOTE: DO NOT MODIFY THIS FUNCTION.
|
116 |
"""
|
117 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
118 |
training_dataset = prepare_dataset()
|
119 |
|
120 |
print("Splitting the dataset...")
|