def RAkELO(dataset_train_x, dataset_train_y, dataset_test_x, dataset_test_y, base_clasif, num_labels, num_models): classifier = RakelO(base_classifier=base_clasif, labelset_size=num_labels, model_count=num_models) classifier.fit(dataset_train_x, dataset_train_y) predictions = classifier.predict(dataset_test_x) Metrics_Accuracy("RAkELO", predictions, dataset_test_y)
def RAkELO(dataset_train_x, dataset_train_y, dataset_test_x, dataset_test_y, base_clasif, num_labels, num_models): classifier = RakelO(base_classifier=base_clasif, labelset_size=num_labels, model_count=num_models) start_time = time.time() classifier.fit(dataset_train_x, dataset_train_y) stop_time = time.time() time_lapsed = stop_time - start_time predictions = classifier.predict(dataset_test_x) Metrics_Accuracy("RAkELO", predictions, dataset_test_y) print("Execution time: {}s".format(time_lapsed))