Skip to content

Instantly share code, notes, and snippets.

View SaschaHeyer's full-sized avatar
👨‍🚀

Sascha Heyer SaschaHeyer

👨‍🚀
View GitHub Profile
@SaschaHeyer
SaschaHeyer / description.py
Created September 21, 2022 18:50
Vertex AI Model Versioining - Description
aiplatform.Model.upload(
version_description="This is a test version description"
)
@SaschaHeyer
SaschaHeyer / version.py
Created September 21, 2022 18:42
Vertex AI Model Versioning - Model Version
aiplatform.Model.upload(
labels={"release": "stable"},
)
@SaschaHeyer
SaschaHeyer / filter.py
Created September 21, 2022 18:30
Vertex AI Model Versioning - Filtering
models = aiplatform.Model.list(filter=f"labels.release=stable")
print("Number of models:", len(models))
print("Version ID:", models[0].version_id)
@SaschaHeyer
SaschaHeyer / remove_models_with_no_traffic.py
Created September 11, 2022 09:00
Vertex AI Model Versioning
for model in endpoint.list_models():
if model.id not in endpoint.traffic_split:
endpoint.undeploy(deployed_model_id = model.id)
@SaschaHeyer
SaschaHeyer / embedding.json
Last active September 5, 2022 19:45
Recommendations System Deep Learning embedding
[0.173280731, -0.180692792, 0.184972495, -0.158537686, 0.0464187413, 0.0330387875, -0.082017608,
0.116319604, 0.114213862, -0.0770902932, 0.140490398, -0.123474516, -0.143807605, -0.16269505,
0.0953937843, 0.00816709, -0.180989, 0.174584985, 0.0205909405, -0.0414723866, -0.0688794553,
-0.101598516, -0.0432119258, 0.0707893446, -0.0506613851, -0.132895559, -0.00127901766, -0.111646101,
0.126078412, 0.13549535, -0.0555192493, 0.196069285, -0.080629319, 0.130173147, -0.145848453,
0.17148535, 0.000497843954, 0.0905586, -0.0872443244, -0.0277212523, 0.0125280414, -0.0683887079,
-0.0402347669, 0.165481687, 0.0783747, -0.000467181439, 0.0971193463, 0.134140953, 0.102389038,
-0.109147869, 0.186242402, 0.158349961, 0.0761855766, -0.187734008, 0.0172080304, 0.159282789,
-0.155050635, -0.0161176361, -0.0204912461, -0.12683472, -0.0215795655, -0.184450388, 0.0941895545,
-0.0449598655]
@SaschaHeyer
SaschaHeyer / model.json
Last active September 5, 2022 17:30
Vertex AI Model Versioning
{
"parentModel": "",
"modelId": "",
"model": { }
}
@SaschaHeyer
SaschaHeyer / upload.py
Created September 5, 2022 14:39
two-tower
model_uploaded = aiplatform.Model.upload(
serving_container_image_uri = "us-central1-docker.pkg.dev/cloud-ml-algos/two-tower/deploy",
)
@SaschaHeyer
SaschaHeyer / train.py
Created August 16, 2022 09:50
Recommendation System train
training_args = ["--training_data_path", TRAINING_DATA_PATH,
"--input_schema_path", INPUT_SCHEMA_PATH,
"--job-dir", OUTPUT_DIR,
"--train_batch_size", str(TRAIN_BATCH_SIZE),
"--num_epochs", str(NUM_EPOCHS)
]
worker_pool_specs = [
{
"machine_spec": {
@SaschaHeyer
SaschaHeyer / schema.json
Created August 16, 2022 09:41
Recommendation System Schema
{
"query":{
"user_id":{
"feature_type":"Id",
"config":{
"num_buckets":91
}
}
},
"candidate":{
@SaschaHeyer
SaschaHeyer / response.json
Created August 7, 2022 07:48
Vertex AI Serving response
{
"predictions": [
{
"confidence": 0.9409326314926147,
"sentiment": "POSITIVE"
},
{
"confidence": 0.9964427351951599,
"sentiment": "NEGATIVE"
}