use of com.google.cloud.aiplatform.v1beta1.DatasetServiceClient in project java-aiplatform by googleapis.
the class ImportDataVideoActionRecognitionSampleTest method setUp.
@Before
public void setUp() throws IOException, InterruptedException, ExecutionException, TimeoutException {
bout = new ByteArrayOutputStream();
out = new PrintStream(bout);
originalPrintStream = System.out;
System.setOut(out);
// create a temp dataset for importing data
DatasetServiceSettings datasetServiceSettings = DatasetServiceSettings.newBuilder().setEndpoint("us-central1-aiplatform.googleapis.com:443").build();
try (DatasetServiceClient datasetServiceClient = DatasetServiceClient.create(datasetServiceSettings)) {
String metadataSchemaUri = "gs://google-cloud-aiplatform/schema/dataset/metadata/video_1.0.0.yaml";
LocationName locationName = LocationName.of(PROJECT, LOCATION);
Dataset dataset = Dataset.newBuilder().setDisplayName("test_dataset_display_name").setMetadataSchemaUri(metadataSchemaUri).build();
OperationFuture<Dataset, CreateDatasetOperationMetadata> datasetFuture = datasetServiceClient.createDatasetAsync(locationName, dataset);
Dataset datasetResponse = datasetFuture.get(300, TimeUnit.SECONDS);
String[] datasetValues = datasetResponse.getName().split("/");
datasetId = datasetValues[datasetValues.length - 1];
}
}
use of com.google.cloud.aiplatform.v1beta1.DatasetServiceClient in project java-aiplatform by googleapis.
the class ImportDataVideoClassificationSampleTest method setUp.
@Before
public void setUp() throws IOException, InterruptedException, ExecutionException, TimeoutException {
bout = new ByteArrayOutputStream();
out = new PrintStream(bout);
originalPrintStream = System.out;
System.setOut(out);
// create a temp dataset for importing data
DatasetServiceSettings datasetServiceSettings = DatasetServiceSettings.newBuilder().setEndpoint("us-central1-aiplatform.googleapis.com:443").build();
try (DatasetServiceClient datasetServiceClient = DatasetServiceClient.create(datasetServiceSettings)) {
String metadataSchemaUri = "gs://google-cloud-aiplatform/schema/dataset/metadata/video_1.0.0.yaml";
LocationName locationName = LocationName.of(PROJECT, LOCATION);
Dataset dataset = Dataset.newBuilder().setDisplayName("test_dataset_display_name").setMetadataSchemaUri(metadataSchemaUri).build();
OperationFuture<Dataset, CreateDatasetOperationMetadata> datasetFuture = datasetServiceClient.createDatasetAsync(locationName, dataset);
Dataset datasetResponse = datasetFuture.get(300, TimeUnit.SECONDS);
String[] datasetValues = datasetResponse.getName().split("/");
datasetId = datasetValues[datasetValues.length - 1];
}
}
use of com.google.cloud.aiplatform.v1beta1.DatasetServiceClient in project java-aiplatform by googleapis.
the class DeleteDatasetSample method deleteDatasetSample.
static void deleteDatasetSample(String project, String datasetId) throws IOException, InterruptedException, ExecutionException, TimeoutException {
DatasetServiceSettings datasetServiceSettings = DatasetServiceSettings.newBuilder().setEndpoint("us-central1-aiplatform.googleapis.com:443").build();
// the "close" method on the client to safely clean up any remaining background resources.
try (DatasetServiceClient datasetServiceClient = DatasetServiceClient.create(datasetServiceSettings)) {
String location = "us-central1";
DatasetName datasetName = DatasetName.of(project, location, datasetId);
OperationFuture<Empty, DeleteOperationMetadata> operationFuture = datasetServiceClient.deleteDatasetAsync(datasetName);
System.out.format("Operation name: %s\n", operationFuture.getInitialFuture().get().getName());
System.out.println("Waiting for operation to finish...");
operationFuture.get(300, TimeUnit.SECONDS);
System.out.format("Deleted Dataset.");
}
}
use of com.google.cloud.aiplatform.v1beta1.DatasetServiceClient in project java-aiplatform by googleapis.
the class ImportDataTextEntityExtractionSample method importDataTextEntityExtractionSample.
static void importDataTextEntityExtractionSample(String project, String datasetId, String gcsSourceUri) throws IOException, InterruptedException, ExecutionException, TimeoutException {
DatasetServiceSettings datasetServiceSettings = DatasetServiceSettings.newBuilder().setEndpoint("us-central1-aiplatform.googleapis.com:443").build();
// the "close" method on the client to safely clean up any remaining background resources.
try (DatasetServiceClient datasetServiceClient = DatasetServiceClient.create(datasetServiceSettings)) {
String location = "us-central1";
String importSchemaUri = "gs://google-cloud-aiplatform/schema/dataset/ioformat/" + "text_extraction_io_format_1.0.0.yaml";
GcsSource.Builder gcsSource = GcsSource.newBuilder();
gcsSource.addUris(gcsSourceUri);
DatasetName datasetName = DatasetName.of(project, location, datasetId);
List<ImportDataConfig> importDataConfigList = Collections.singletonList(ImportDataConfig.newBuilder().setGcsSource(gcsSource).setImportSchemaUri(importSchemaUri).build());
OperationFuture<ImportDataResponse, ImportDataOperationMetadata> importDataResponseFuture = datasetServiceClient.importDataAsync(datasetName, importDataConfigList);
System.out.format("Operation name: %s\n", importDataResponseFuture.getInitialFuture().get().getName());
System.out.println("Waiting for operation to finish...");
ImportDataResponse importDataResponse = importDataResponseFuture.get(300, TimeUnit.SECONDS);
System.out.format("Import Data Text Entity Extraction Response: %s\n", importDataResponse.toString());
}
}
use of com.google.cloud.aiplatform.v1beta1.DatasetServiceClient in project java-aiplatform by googleapis.
the class ImportDataVideoActionRecognitionSample method importDataVideoActionRecognitionSample.
static void importDataVideoActionRecognitionSample(String project, String datasetId, String gcsSourceUri) throws IOException, ExecutionException, InterruptedException {
DatasetServiceSettings settings = DatasetServiceSettings.newBuilder().setEndpoint("us-central1-aiplatform.googleapis.com:443").build();
String location = "us-central1";
// the "close" method on the client to safely clean up any remaining background resources.
try (DatasetServiceClient client = DatasetServiceClient.create(settings)) {
GcsSource gcsSource = GcsSource.newBuilder().addUris(gcsSourceUri).build();
ImportDataConfig importConfig0 = ImportDataConfig.newBuilder().setGcsSource(gcsSource).setImportSchemaUri("gs://google-cloud-aiplatform/schema/dataset/ioformat/" + "video_action_recognition_io_format_1.0.0.yaml").build();
List<ImportDataConfig> importConfigs = new ArrayList<>();
importConfigs.add(importConfig0);
DatasetName name = DatasetName.of(project, location, datasetId);
OperationFuture<ImportDataResponse, ImportDataOperationMetadata> response = client.importDataAsync(name, importConfigs);
// You can use OperationFuture.getInitialFuture to get a future representing the initial
// response to the request, which contains information while the operation is in progress.
System.out.format("Operation name: %s\n", response.getInitialFuture().get().getName());
// OperationFuture.get() will block until the operation is finished.
ImportDataResponse importDataResponse = response.get();
System.out.format("importDataResponse: %s\n", importDataResponse);
}
}
Aggregations