Skip to content
Go to file
Cannot retrieve contributors at this time
94 lines (55 sloc) 6.88 KB

K Means Tutorial

This tutorial describes how to perform a K-Means analysis. By the end of this tutorial the user should know how to specify, run, and interpret a K-means model in H2O using Flow.

Those who have never used H2O before should refer to Getting Started for additional instructions on how to run H2O Flow.

In the latest version of H2O, the K-means algorithm has a "k-modes" function that allows you to use mixed categorical and real-valued data. By using dissimilarity measures to handle categoricals, replacing cluster means with cluster modes, and using a frequency-based method to update modes in the clustering process to minimize the clustering costs, the k-modes algorithm is scalable in both the number of clusters and the number of records. The k-modes method is used anytime categorical data is present.

For more information, refer to "A Fast Clustering Algorithm to Cluster Very Large Categorical Data Sets in Data Mining" and "Extensions to the k-Means Algorithm for Clustering Large Data Sets with Catgorical Values" by Zhexue Huang.

Getting Started

First get the tutorial's data set from here (on the next page select File > Save to download the text file). (This data originally comes from the publicly available data set, which can be found at In the tutorial's link above we've removed random spacing that causes the data set to look like 9 columns instead of 8.)

The data are composed of 210 observations, 7 attributes, and an a priori grouping assignment. All data are positively valued and continuous.

If you don't have any data of your own to work with, you can find some example datasets at

Importing Data

Before creating a model, import data into H2O:

  1. Click the Assist Me! button (the last button in the row of buttons below the menus).

Assist Me button

  1. Click the importFiles link and enter the file path to the dataset in the Search entry field.
  2. Click the Add all link to add the file to the import queue, then click the Import button. Importing Files

Parsing Data

Now, parse the imported data:

  1. Click the Parse these files... button.

    Note: The default options typically do not need to be changed unless the data does not parse correctly.

  2. From the drop-down Parser list, select the file type of the data set (Auto, XLS, CSV, or SVMLight).

  3. If the data uses a separator, select it from the drop-down Separator list.

  4. If the data uses a column header as the first row, select the First row contains column names radio button. If the first row contains data, select the First row contains data radio button. You can also select the Auto radio button to have H2O automatically determine if the first row of the dataset contains the column names or data.

  5. If the data uses apostrophes ( ' - also known as single quotes), check the Enable single quotes as a field quotation character checkbox.

  6. Review the data in the Edit Column Names and Types section, then click the Parse button.

Parsing Data

NOTE: Make sure the parse is complete by confirming progress is 100% before continuing to the next step, model building. For small datasets, this should only take a few seconds, but larger datasets take longer to parse.

Building a Model

  1. Once data are parsed, click the View button, then click the Build Model button.

  2. Select K-means from the drop-down Select an algorithm menu, then click the Build model button.

  3. If the parsed seeds_dataset.hex file is not already listed in the Training_frame drop-down list, select it. Otherwise, continue to the next step.

  4. From the Ignored_columns section, select the columns to ignore in the Available area to move them to the Selected area. For this example, select column 8 (the a priori known clusters for this dataset).

  5. In the K field, specify the number of clusters. For this example, enter 3.

  6. In the Max_iterations field, specify the maximum number of iterations. For this example, enter 100.

  7. From the drop-down Init menu, select the initialization mode. For this example, select PlusPlus.

    • Random initialization randomly samples the k-specified value of the rows of the training data as cluster centers.

    • PlusPlus initialization chooses one initial center at random and weights the random selection of subsequent centers so that points furthest from the first center are more likely to be chosen.

    • Furthest initialization chooses one initial center at random and then chooses the next center to be the point furthest away in terms of Euclidean distance.

    • User initialization requires the corresponding User_points parameter. To define a specific initialization point, select the imported dataset .hex file from the drop-down User_points list, then select User from the drop-down Init list.

      Note: The user-specified points dataset must have the same number of columns as the training dataset.

  8. Uncheck the Standardize checkbox to disable column standardization.

  9. Click the Build Model button.

K-Means Model Builder cell

K-Means Output

K-Means randomly chooses starting points and converges on optimal centroids. The cluster number is arbitrary, and should be thought of as a factor.

The output for K-Means includes the following:

  • Model parameters (hidden)
  • A graph of the scoring history (number of iterations vs. average within the cluster's sum of squares)
  • Output (model category, validation metrics if applicable, and centers std)
  • Model Summary (number of clusters, number of categorical columns, number of iterations, avg. within sum of squares, avg. sum of squares, avg. between the sum of squares)
  • Scoring history (number of iterations, avg. change of standardized centroids, avg. within cluster sum of squares)
  • Training metrics (model name, checksum name, frame name, frame checksum name, description if applicable, model category, duration in ms, scoring time, predictions, MSE, avg. within sum of squares, avg. between sum of squares)
  • Centroid statistics (centroid number, size, within sum of squares)
  • Cluster means (centroid number, column)
  • Preview POJO
You can’t perform that action at this time.