Sha256: 3755bec4ff82dfb3718671631f3af07a6e17ad1841c24fced299ddcf2443cf5c
Contents?: true
Size: 1.28 KB
Versions: 19
Compression:
Stored size: 1.28 KB
Contents
# frozen_string_literal: true
require 'rumale/base/evaluator'
module Rumale
module EvaluationMeasure
# Purity is a class that calculates the purity of cluatering results.
#
# @example
# evaluator = Rumale::EvaluationMeasure::Purity.new
# puts evaluator.score(ground_truth, predicted)
#
# *Reference*
# - C D. Manning, P. Raghavan, and H. Schutze, "Introduction to Information Retrieval," Cambridge University Press., 2008.
class Purity
include Base::Evaluator
# Calculate purity
#
# @param y_true [Numo::Int32] (shape: [n_samples]) Ground truth labels.
# @param y_pred [Numo::Int32] (shape: [n_samples]) Predicted cluster labels.
# @return [Float] Purity
def score(y_true, y_pred)
y_true = check_convert_label_array(y_true)
y_pred = check_convert_label_array(y_pred)
# initiazlie some variables.
purity = 0
n_samples = y_pred.size
class_ids = y_true.to_a.uniq
cluster_ids = y_pred.to_a.uniq
# calculate purity.
cluster_ids.each do |k|
pr_sample_ids = y_pred.eq(k).where.to_a
purity += class_ids.map { |j| (pr_sample_ids & y_true.eq(j).where.to_a).size }.max
end
purity.fdiv(n_samples)
end
end
end
end
Version data entries
19 entries across 19 versions & 1 rubygems