зеркало из https://github.com/mozilla/bugbug.git
Родитель
7dee06e724
Коммит
f50971f7f9
3
run.py
3
run.py
|
@ -17,11 +17,11 @@ from sklearn.pipeline import FeatureUnion
|
|||
from sklearn.pipeline import Pipeline
|
||||
|
||||
import bug_features
|
||||
from get_bugs import get_bug_fields
|
||||
from get_bugs import get_bugs
|
||||
from get_bugs import get_labels
|
||||
from utils import ItemSelector
|
||||
|
||||
|
||||
def go():
|
||||
# Get labels.
|
||||
classes = get_labels()
|
||||
|
@ -135,5 +135,6 @@ def go():
|
|||
print('Recall: {}'.format(metrics.recall_score(y_test, y_pred)))
|
||||
print(metrics.confusion_matrix(y_test, y_pred))
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
go()
|
||||
|
|
Загрузка…
Ссылка в новой задаче