aaa / data /artifacts /models /recommender /training_config.json
work-sejal
Add models and dataset
8ece6f3
Raw
History Blame Contribute Delete
763 Bytes
{
"model_name": "recommender",
"model_version": "recommender_v2_baseline_001",
"dataset_version": "2.0.0",
"seed": 42,
"split_counts": {
"train": 5671,
"validation": 1214,
"test": 1215
},
"hyperparameters": {
"n_estimators": 100,
"max_depth": 4,
"random_state": 42,
"algorithm": "GradientBoostingClassifier",
"encoder": "OrdinalEncoder"
},
"feature_columns": [
"priority",
"ai_confidence",
"recommendation_type",
"grade",
"subject"
],
"categorical_columns": [
"priority",
"recommendation_type",
"subject"
],
"numeric_columns": [
"ai_confidence",
"grade"
],
"target_columns": [
"clicked",
"is_completed"
],
"algorithm": "GradientBoostingClassifier"
}