Search in sources :

Example 1 with TablesModelColumnInfo

use of com.google.cloud.automl.v1beta1.TablesModelColumnInfo in project java-automl by googleapis.

the class TablesGetModel method getModel.

// Demonstrates using the AutoML client to get model details.
public static void getModel(String projectId, String computeRegion, String modelId) throws IOException, StatusRuntimeException {
    // the "close" method on the client to safely clean up any remaining background resources.
    try (AutoMlClient client = AutoMlClient.create()) {
        // Get the full path of the model.
        ModelName modelFullId = ModelName.of(projectId, computeRegion, modelId);
        // Get complete detail of the model.
        Model model = client.getModel(modelFullId);
        // Display the model information.
        System.out.format("Model name: %s%n", model.getName());
        System.out.format("Model Id: %s\n", model.getName().split("/")[model.getName().split("/").length - 1]);
        System.out.format("Model display name: %s%n", model.getDisplayName());
        System.out.format("Dataset Id: %s%n", model.getDatasetId());
        System.out.println("Tables Model Metadata: ");
        System.out.format("\tTraining budget: %s%n", model.getTablesModelMetadata().getTrainBudgetMilliNodeHours());
        System.out.format("\tTraining cost: %s%n", model.getTablesModelMetadata().getTrainBudgetMilliNodeHours());
        DateFormat dateFormat = new SimpleDateFormat("yyyy-MM-dd'T'HH:mm:ss.SSSZ");
        String createTime = dateFormat.format(new java.util.Date(model.getCreateTime().getSeconds() * 1000));
        System.out.format("Model create time: %s%n", createTime);
        System.out.format("Model deployment state: %s%n", model.getDeploymentState());
        // Get features of top importance
        for (TablesModelColumnInfo info : model.getTablesModelMetadata().getTablesModelColumnInfoList()) {
            System.out.format("Column: %s - Importance: %.2f%n", info.getColumnDisplayName(), info.getFeatureImportance());
        }
    }
}
Also used : TablesModelColumnInfo(com.google.cloud.automl.v1beta1.TablesModelColumnInfo) ModelName(com.google.cloud.automl.v1beta1.ModelName) SimpleDateFormat(java.text.SimpleDateFormat) DateFormat(java.text.DateFormat) Model(com.google.cloud.automl.v1beta1.Model) SimpleDateFormat(java.text.SimpleDateFormat) AutoMlClient(com.google.cloud.automl.v1beta1.AutoMlClient)

Aggregations

AutoMlClient (com.google.cloud.automl.v1beta1.AutoMlClient)1 Model (com.google.cloud.automl.v1beta1.Model)1 ModelName (com.google.cloud.automl.v1beta1.ModelName)1 TablesModelColumnInfo (com.google.cloud.automl.v1beta1.TablesModelColumnInfo)1 DateFormat (java.text.DateFormat)1 SimpleDateFormat (java.text.SimpleDateFormat)1