datasetId,

modelDisplayName,

trainingPipelineDisplayName,

project,

location = 'us-central1'

) {

// [START aiplatform_create_training_pipeline_video_classification_sample]

/**

* TODO(developer): Uncomment these variables before running the sample.\

* (Not necessary if passing values as arguments)

*/

// const datasetId = 'YOUR_DATASET_ID';

// const modelDisplayName = 'YOUR_MODEL_DISPLAY_NAME';

// const trainingPipelineDisplayName = 'YOUR_TRAINING_PIPELINE_DISPLAY_NAME';

// const project = 'YOUR_PROJECT_ID';

// const location = 'YOUR_PROJECT_LOCATION';

const aiplatform = require('@google-cloud/aiplatform');

const {definition} =

aiplatform.protos.google.cloud.aiplatform.v1.schema.trainingjob;

// Imports the Google Cloud Pipeline Service Client library

const {PipelineServiceClient} = aiplatform.v1;

// Specifies the location of the api endpoint

const clientOptions = {

apiEndpoint: 'us-central1-aiplatform.googleapis.com',

};

// Instantiates a client

const pipelineServiceClient = new PipelineServiceClient(clientOptions);

async function createTrainingPipelineVideoClassification() {

// Configure the parent resource

const parent = `projects/${project}/locations/${location}`;

// Values should match the input expected by your model.

const trainingTaskInputObj = new definition.AutoMlVideoClassificationInputs(

{}

);

const trainingTaskInputs = trainingTaskInputObj.toValue();

const modelToUpload = {displayName: modelDisplayName};

const inputDataConfig = {datasetId: datasetId};

const trainingPipeline = {

displayName: trainingPipelineDisplayName,

trainingTaskDefinition:

'gs://google-cloud-aiplatform/schema/trainingjob/definition/automl_video_classification_1.0.0.yaml',

trainingTaskInputs,

inputDataConfig,

modelToUpload,

};

const request = {

parent,

trainingPipeline,

};

// Create training pipeline request

const [response] =

await pipelineServiceClient.createTrainingPipeline(request);

console.log('Create training pipeline video classification response');

console.log(`Name : ${response.name}`);

console.log('Raw response:');

console.log(JSON.stringify(response, null, 2));

}

createTrainingPipelineVideoClassification();

// [END aiplatform_create_training_pipeline_video_classification_sample]