Skip to content

Commit ff9107e

Browse files
committed
created separate generate_data function
Signed-off-by: Amit Sharma <[email protected]>
1 parent eded1e3 commit ff9107e

File tree

2 files changed

+5
-13
lines changed

2 files changed

+5
-13
lines changed

pywhyllm/datasets/dataset.py

Lines changed: 1 addition & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@ class Dataset(Protocol):
44

55
def graph(self):
66
"""
7-
Output the graph associated with the dataset.
7+
Output the ground truth graph associated with the dataset.
88
99
Returns:
1010
A networkx graph
@@ -20,11 +20,3 @@ def data(self):
2020
"""
2121
pass
2222

23-
def generate_data(self):
24-
"""
25-
Generate synthetic data based on the dataset's causal graph.
26-
27-
Returns:
28-
A numpy array containing the synthetic data
29-
"""
30-
pass

pywhyllm/datasets/metrics.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,25 +1,25 @@
11

22

33
def accuracy(edges, true_edges):
4-
"""Accuracy metric for predicted edges
4+
"""Accuracy metric for predicted edges for a given dataset
55
"""
66
pass
77

88

99
def precision(edges, true_edges):
10-
"""Precision metric for predicted edges
10+
"""Precision metric for predicted edges for a given dataset
1111
"""
1212
pass
1313

1414

1515
def recall(edges, true_edges):
16-
"""Recall metric for predicted edges
16+
"""Recall metric for predicted edges for a given dataset
1717
"""
1818
pass
1919

2020

2121
def F1(edges, true_edges):
22-
"""F1 metric for predicted edges
22+
"""F1 metric for predicted edges for a given dataset
2323
"""
2424
pass
2525

0 commit comments

Comments
 (0)