发布于 2022-10-06 17:36:21
您需要将hypertune
添加到笔记本中,以便它能够在界面上编写不同的超参数及其性能:
import hypertune
hp_metric = f1_score(y_test, y_pred, average='weighted')
hpt = hypertune.HyperTune()
hpt.report_hyperparameter_tuning_metric(hyperparameter_metric_tag='accuracy',metric_value=hp_metric,global_step=100)
还添加了args来调整模型:
if __name__ == '__main__':
parser = argparse.ArgumentParser()
# Input Arguments
parser.add_argument(
'--max_depth',
help = 'RF model parameter- depth',
type = int,
default = 100
)
parser.add_argument(
'--max_features',
help = 'RF model parameter- Features',
type = int,
default = 34
)
parser.add_argument(
'--max_leaf_nodes',
help = 'RF max_leaf_nodes',
type = int,
default = 8
)
parser.add_argument(
'--min_samples_leaf',
help = 'RF min_samples_leaf',
type = int,
default = 1
)
https://stackoverflow.com/questions/73977843
复制相似问题