in src/responsibleai/rai_analyse/create_error_analysis.py [0:0]
def main(args):
my_run = Run.get_context()
# Load the RAI Insights object
rai_i: RAIInsights = create_rai_insights_from_port_path(
my_run, args.rai_insights_dashboard
)
# Add the error analysis
rai_i.error_analysis.add(
max_depth=args.max_depth,
num_leaves=args.num_leaves,
min_child_samples=args.min_child_samples,
filter_features=args.filter_features,
)
_logger.info("Added error analysis")
# Compute
rai_i.compute()
_logger.info("Computation complete")
# Save
save_to_output_port(rai_i, args.error_analysis_path, RAIToolType.ERROR_ANALYSIS)
_logger.info("Saved to output port")
# Copy the dashboard info file
copy_dashboard_info_file(args.rai_insights_dashboard, args.error_analysis_path)
_logger.info("Completing")