Class: AprendizajeMaquina::KmeansClustering

Inherits:
Object
  • Object
show all
Defined in:
lib/aprendizaje_maquina/clustering.rb

Instance Method Summary collapse

Constructor Details

#initialize(num_of_cluster_centroids, dataset_matrix) ⇒ KmeansClustering

Returns a new instance of KmeansClustering.


3
4
5
6
7
8
9
 
# File 'lib/aprendizaje_maquina/clustering.rb', line 3

def initialize(num_of_cluster_centroids,dataset_matrix)
	@num_of_cluster_centroids = num_of_cluster_centroids
	@dataset_matrix = dataset_matrix
	@num_columns = @dataset_matrix.column_count
	@num_rows = @dataset_matrix.row_count
	@cluster_centroids = init_cluster_centroids
end

Instance Method Details

#cluster(num) ⇒ Object 


15
16
17
 
# File 'lib/aprendizaje_maquina/clustering.rb', line 15

def cluster(num)
	get("@cluster_#{num}")
end

#fit(iterations) ⇒ Object 


11
12
13
 
# File 'lib/aprendizaje_maquina/clustering.rb', line 11

def fit(iterations)
	clustering(iterations)
end

#predict(vector) ⇒ Object 


19
20
21
22
23
24
25
26
 
# File 'lib/aprendizaje_maquina/clustering.rb', line 19

def predict(vector)	
	array = []			
	@cluster_centroids.each do |cluster|
	  array << (vector-cluster).r
	end
	cluster = array.index(array.min)
	cluster
end