|
@@ -75,8 +75,8 @@ def algo_comparison(algo1, algo2, algo1_name, algo2_name):
|
|
|
for col in range(2):
|
|
|
i = row + 2 * col
|
|
|
|
|
|
- for corpus in set(data["corpus"]):
|
|
|
- mask = np.array([data["corpus"][k - 1] == corpus for k in data["children"]])
|
|
|
+ for corpus in set(algo1["corpus"]):
|
|
|
+ mask = np.array([algo1["corpus"][k - 1] == corpus for k in algo1["children"]])
|
|
|
vocs_algo1 = algo1["vocs"][mask,i]
|
|
|
vocs_algo2 = algo2["vocs"][mask,i]
|
|
|
|