Spaces:
Sleeping
Sleeping
File size: 1,727 Bytes
07c0591 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 |
from flask import jsonify
def get_initial_data():
initial_nodes = [
{
"id": "data-input",
"position": {"x": 0, "y": 0},
"data": {"label": "Data Input"}
},
{
"id": "data-preprocessing",
"position": {"x": 200, "y": 0},
"data": {"label": "Data Preprocessing"}
},
{
"id": "model-training",
"position": {"x": 400, "y": 0},
"data": {"label": "Model Training"}
},
{
"id": "model-evaluation",
"position": {"x": 0, "y": 200},
"data": {"label": "Model Evaluation"}
},
{
"id": "prediction",
"position": {"x": 200, "y": 200},
"data": {"label": "Prediction"}
},
{
"id": "data-visualization",
"position": {"x": 400, "y": 200},
"data": {"label": "Data Visualization"}
},
]
initial_edges = [
{"id": "data-input-to-preprocessing", "source": "data-input", "target": "data-preprocessing"},
{"id": "preprocessing-to-training", "source": "data-preprocessing", "target": "model-training"},
{"id": "training-to-evaluation", "source": "model-training", "target": "model-evaluation"},
{"id": "training-to-prediction", "source": "model-training", "target": "prediction"},
{"id": "evaluation-to-visualization", "source": "model-evaluation", "target": "data-visualization"},
{"id": "prediction-to-visualization", "source": "prediction", "target": "data-visualization"}
]
return jsonify({
"initialNodes": initial_nodes,
"initialEdges": initial_edges
})
|