Spaces:
Configuration error
Configuration error
Update tasks.py
Browse files
tasks.py
CHANGED
|
@@ -1,21 +1,37 @@
|
|
| 1 |
from invoke import task
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 2 |
|
| 3 |
@task
|
| 4 |
def preprocess(ctx):
|
| 5 |
print("Running preprocess step...")
|
| 6 |
# Add invoke logic for preprocessing
|
|
|
|
|
|
|
| 7 |
|
| 8 |
@task
|
| 9 |
def train(ctx):
|
| 10 |
print("Running training step...")
|
| 11 |
# Add invoke logic for training
|
|
|
|
|
|
|
|
|
|
|
|
|
| 12 |
|
| 13 |
-
@task
|
| 14 |
-
def predict(ctx):
|
| 15 |
-
print("Running prediction step...")
|
| 16 |
-
# Add invoke logic for prediction
|
| 17 |
|
| 18 |
@task
|
| 19 |
def analyze(ctx):
|
| 20 |
print("Running analysis step...")
|
| 21 |
-
# Add invoke logic for result analysis
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
from invoke import task
|
| 2 |
+
import preprocess
|
| 3 |
+
import train
|
| 4 |
+
import results
|
| 5 |
+
import predict
|
| 6 |
+
|
| 7 |
|
| 8 |
@task
|
| 9 |
def preprocess(ctx):
|
| 10 |
print("Running preprocess step...")
|
| 11 |
# Add invoke logic for preprocessing
|
| 12 |
+
preprocess.run_preprocessing()
|
| 13 |
+
|
| 14 |
|
| 15 |
@task
|
| 16 |
def train(ctx):
|
| 17 |
print("Running training step...")
|
| 18 |
# Add invoke logic for training
|
| 19 |
+
best_params = train.run_training(max_depth=10, min_samples_split=2)
|
| 20 |
+
print(f"Best Parameters: {best_params}")
|
| 21 |
+
|
| 22 |
+
analyze(ctx)
|
| 23 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 24 |
|
| 25 |
@task
|
| 26 |
def analyze(ctx):
|
| 27 |
print("Running analysis step...")
|
| 28 |
+
# Add invoke logic for result analysis
|
| 29 |
+
results.evaluate_results()
|
| 30 |
+
|
| 31 |
+
|
| 32 |
+
@task
|
| 33 |
+
def predict(ctx):
|
| 34 |
+
print("Running prediction step...")
|
| 35 |
+
# Add invoke logic for prediction
|
| 36 |
+
predict.run_prediction()
|
| 37 |
+
|