diff --git a/onnxmltools/convert/xgboost/operator_converters/XGBoost.py b/onnxmltools/convert/xgboost/operator_converters/XGBoost.py index cb43ca1f..6f3fdc2e 100644 --- a/onnxmltools/convert/xgboost/operator_converters/XGBoost.py +++ b/onnxmltools/convert/xgboost/operator_converters/XGBoost.py @@ -75,7 +75,7 @@ def _add_node(attr_pairs, is_classifier, tree_id, tree_weight, node_id, feature_id)) else: try: - feature_id = int(feature_id) + feature_id = int(float(feature_id)) except ValueError: raise RuntimeError( "Unable to interpret '{0}', feature "