55 < meta name ="color-scheme " content ="light dark "> < link rel ="index " title ="Index " href ="../../../genindex.html "> < link rel ="search " title ="Search " href ="../../../search.html ">
66
77 <!-- Generated with Sphinx 9.0.4 and Furo 2025.12.19 -->
8- < title > spflow.interfaces.sklearn - SPFlow 1.0.0 </ title >
8+ < title > spflow.interfaces.sklearn - SPFlow 1.0.2 </ title >
99 < link rel ="stylesheet " type ="text/css " href ="../../../_static/pygments.css?v=d111a655 " />
1010 < link rel ="stylesheet " type ="text/css " href ="../../../_static/styles/furo.css?v=7bdb33bb " />
1111 < link rel ="stylesheet " type ="text/css " href ="../../../_static/copybutton.css?v=76b2166b " />
167167 </ label >
168168 </ div >
169169 < div class ="header-center ">
170- < a href ="../../../index.html "> < div class ="brand "> SPFlow 1.0.0 </ div > </ a >
170+ < a href ="../../../index.html "> < div class ="brand "> SPFlow 1.0.2 </ div > </ a >
171171 </ div >
172172 < div class ="header-right ">
173173 < div class ="theme-toggle-container theme-toggle-header ">
188188
189189 < div class ="sidebar-sticky "> < a class ="sidebar-brand " href ="../../../index.html ">
190190
191- < span class ="sidebar-brand-text "> SPFlow 1.0.0 </ span >
191+ < span class ="sidebar-brand-text "> SPFlow 1.0.2 </ span >
192192
193193</ a > < form class ="sidebar-search-container " method ="get " action ="../../../search.html " role ="search ">
194194 < input class ="sidebar-search " placeholder ="Search " name ="q " aria-label ="Search ">
@@ -344,6 +344,14 @@ <h1>Source code for spflow.interfaces.sklearn</h1><div class="highlight"><pre>
344344 < span class ="k "> raise</ span > < span class ="n "> InvalidParameterError</ span > < span class ="p "> (</ span > < span class ="sa "> f</ span > < span class ="s2 "> "Unknown dtype '</ span > < span class ="si "> {</ span > < span class ="n "> dtype</ span > < span class ="si "> }</ span > < span class ="s2 "> '. Use 'float32', 'float64', or None."</ span > < span class ="p "> )</ span >
345345
346346
347+ < span class ="k "> def</ span > < span class ="w "> </ span > < span class ="nf "> _default_torch_device</ span > < span class ="p "> ()</ span > < span class ="o "> -></ span > < span class ="n "> torch</ span > < span class ="o "> .</ span > < span class ="n "> device</ span > < span class ="p "> :</ span >
348+ < span class ="w "> </ span > < span class ="sd "> """Return the active default torch device, falling back to CPU when unavailable."""</ span >
349+ < span class ="n "> get_default_device</ span > < span class ="o "> =</ span > < span class ="nb "> getattr</ span > < span class ="p "> (</ span > < span class ="n "> torch</ span > < span class ="p "> ,</ span > < span class ="s2 "> "get_default_device"</ span > < span class ="p "> ,</ span > < span class ="kc "> None</ span > < span class ="p "> )</ span >
350+ < span class ="k "> if</ span > < span class ="n "> get_default_device</ span > < span class ="ow "> is</ span > < span class ="kc "> None</ span > < span class ="p "> :</ span >
351+ < span class ="k "> return</ span > < span class ="n "> torch</ span > < span class ="o "> .</ span > < span class ="n "> device</ span > < span class ="p "> (</ span > < span class ="s2 "> "cpu"</ span > < span class ="p "> )</ span >
352+ < span class ="k "> return</ span > < span class ="n "> torch</ span > < span class ="o "> .</ span > < span class ="n "> device</ span > < span class ="p "> (</ span > < span class ="n "> get_default_device</ span > < span class ="p "> ())</ span >
353+
354+
347355< span class ="k "> def</ span > < span class ="w "> </ span > < span class ="nf "> _reduce_log_likelihood</ span > < span class ="p "> (</ span >
348356 < span class ="n "> ll</ span > < span class ="p "> :</ span > < span class ="n "> torch</ span > < span class ="o "> .</ span > < span class ="n "> Tensor</ span > < span class ="p "> ,</ span >
349357 < span class ="o "> *</ span > < span class ="p "> ,</ span >
@@ -412,7 +420,8 @@ <h1>Source code for spflow.interfaces.sklearn</h1><div class="highlight"><pre>
412420< span class ="sd "> leaf_out_channels: Output channels for the leaf module template (passed to `Normal`).</ span >
413421< span class ="sd "> min_instances_slice: Stopping criterion for structure learning (forwarded if not overridden).</ span >
414422< span class ="sd "> min_features_slice: Stopping criterion for structure learning (forwarded if not overridden).</ span >
415- < span class ="sd "> device: Torch device string (e.g., "cpu", "cuda"). If None, uses model device or CPU.</ span >
423+ < span class ="sd "> device: Torch device string (e.g., "cpu", "cuda"). If None, uses model device or the</ span >
424+ < span class ="sd "> active PyTorch default device.</ span >
416425< span class ="sd "> dtype: Torch dtype string ("float32", "float64") for inputs.</ span >
417426< span class ="sd "> channel_agg: How to aggregate multiple output channels into a scalar log-likelihood.</ span >
418427< span class ="sd "> repetition_agg: How to aggregate multiple repetitions into a scalar log-likelihood.</ span >
@@ -455,7 +464,7 @@ <h1>Source code for spflow.interfaces.sklearn</h1><div class="highlight"><pre>
455464 < span class ="k "> return</ span > < span class ="bp "> self</ span > < span class ="o "> .</ span > < span class ="n "> model_</ span > < span class ="o "> .</ span > < span class ="n "> device</ span >
456465 < span class ="k "> if</ span > < span class ="bp "> self</ span > < span class ="o "> .</ span > < span class ="n "> model</ span > < span class ="ow "> is</ span > < span class ="ow "> not</ span > < span class ="kc "> None</ span > < span class ="p "> :</ span >
457466 < span class ="k "> return</ span > < span class ="bp "> self</ span > < span class ="o "> .</ span > < span class ="n "> model</ span > < span class ="o "> .</ span > < span class ="n "> device</ span >
458- < span class ="k "> return</ span > < span class ="n "> torch </ span > < span class ="o " > . </ span > < span class =" n " > device </ span > < span class =" p "> (</ span > < span class =" s2 " > "cpu" </ span > < span class =" p " > )</ span >
467+ < span class ="k "> return</ span > < span class ="n "> _default_torch_device </ span > < span class ="p "> ()</ span >
459468
460469 < span class ="k "> def</ span > < span class ="w "> </ span > < span class ="nf "> _to_tensor</ span > < span class ="p "> (</ span > < span class ="bp "> self</ span > < span class ="p "> ,</ span > < span class ="n "> x</ span > < span class ="p "> :</ span > < span class ="n "> Any</ span > < span class ="p "> )</ span > < span class ="o "> -></ span > < span class ="n "> torch</ span > < span class ="o "> .</ span > < span class ="n "> Tensor</ span > < span class ="p "> :</ span >
461470 < span class ="n "> arr</ span > < span class ="o "> =</ span > < span class ="n "> _as_2d_numpy</ span > < span class ="p "> (</ span > < span class ="n "> x</ span > < span class ="p "> )</ span >
@@ -465,7 +474,9 @@ <h1>Source code for spflow.interfaces.sklearn</h1><div class="highlight"><pre>
465474 < span class ="k "> def</ span > < span class ="w "> </ span > < span class ="nf "> _leaf_template</ span > < span class ="p "> (</ span > < span class ="bp "> self</ span > < span class ="p "> ,</ span > < span class ="n "> n_features</ span > < span class ="p "> :</ span > < span class ="nb "> int</ span > < span class ="p "> )</ span > < span class ="o "> -></ span > < span class ="n "> Any</ span > < span class ="p "> :</ span >
466475 < span class ="k "> if</ span > < span class ="bp "> self</ span > < span class ="o "> .</ span > < span class ="n "> leaf</ span > < span class ="o "> !=</ span > < span class ="s2 "> "normal"</ span > < span class ="p "> :</ span >
467476 < span class ="k "> raise</ span > < span class ="n "> InvalidParameterError</ span > < span class ="p "> (</ span > < span class ="sa "> f</ span > < span class ="s2 "> "Unknown leaf '</ span > < span class ="si "> {</ span > < span class ="bp "> self</ span > < span class ="o "> .</ span > < span class ="n "> leaf</ span > < span class ="si "> }</ span > < span class ="s2 "> '."</ span > < span class ="p "> )</ span >
468- < span class ="k "> return</ span > < span class ="n "> Normal</ span > < span class ="p "> (</ span > < span class ="n "> scope</ span > < span class ="o "> =</ span > < span class ="n "> Scope</ span > < span class ="p "> (</ span > < span class ="nb "> list</ span > < span class ="p "> (</ span > < span class ="nb "> range</ span > < span class ="p "> (</ span > < span class ="n "> n_features</ span > < span class ="p "> ))),</ span > < span class ="n "> out_channels</ span > < span class ="o "> =</ span > < span class ="bp "> self</ span > < span class ="o "> .</ span > < span class ="n "> leaf_out_channels</ span > < span class ="p "> )</ span >
477+ < span class ="k "> return</ span > < span class ="n "> Normal</ span > < span class ="p "> (</ span > < span class ="n "> scope</ span > < span class ="o "> =</ span > < span class ="n "> Scope</ span > < span class ="p "> (</ span > < span class ="nb "> list</ span > < span class ="p "> (</ span > < span class ="nb "> range</ span > < span class ="p "> (</ span > < span class ="n "> n_features</ span > < span class ="p "> ))),</ span > < span class ="n "> out_channels</ span > < span class ="o "> =</ span > < span class ="bp "> self</ span > < span class ="o "> .</ span > < span class ="n "> leaf_out_channels</ span > < span class ="p "> )</ span > < span class ="o "> .</ span > < span class ="n "> to</ span > < span class ="p "> (</ span >
478+ < span class ="bp "> self</ span > < span class ="o "> .</ span > < span class ="n "> _device</ span > < span class ="p "> ()</ span >
479+ < span class ="p "> )</ span >
469480
470481 < span class ="k "> def</ span > < span class ="w "> </ span > < span class ="nf "> _structure_spec</ span > < span class ="p "> (</ span > < span class ="bp "> self</ span > < span class ="p "> )</ span > < span class ="o "> -></ span > < span class ="n "> _StructureLearnerSpec</ span > < span class ="p "> :</ span >
471482 < span class ="k "> if</ span > < span class ="bp "> self</ span > < span class ="o "> .</ span > < span class ="n "> structure_learner</ span > < span class ="ow "> not</ span > < span class ="ow "> in</ span > < span class ="p "> (</ span > < span class ="s2 "> "learn_spn"</ span > < span class ="p "> ,</ span > < span class ="s2 "> "prometheus"</ span > < span class ="p "> ):</ span >
@@ -602,7 +613,7 @@ <h1>Source code for spflow.interfaces.sklearn</h1><div class="highlight"><pre>
602613 < span class ="k "> return</ span > < span class ="n "> torch</ span > < span class ="o "> .</ span > < span class ="n "> device</ span > < span class ="p "> (</ span > < span class ="bp "> self</ span > < span class ="o "> .</ span > < span class ="n "> device</ span > < span class ="p "> )</ span >
603614 < span class ="k "> if</ span > < span class ="nb "> hasattr</ span > < span class ="p "> (</ span > < span class ="bp "> self</ span > < span class ="o "> .</ span > < span class ="n "> model</ span > < span class ="p "> ,</ span > < span class ="s2 "> "device"</ span > < span class ="p "> ):</ span >
604615 < span class ="k "> return</ span > < span class ="n "> torch</ span > < span class ="o "> .</ span > < span class ="n "> device</ span > < span class ="p "> (</ span > < span class ="nb "> getattr</ span > < span class ="p "> (</ span > < span class ="bp "> self</ span > < span class ="o "> .</ span > < span class ="n "> model</ span > < span class ="p "> ,</ span > < span class ="s2 "> "device"</ span > < span class ="p "> ))</ span >
605- < span class ="k "> return</ span > < span class ="n "> torch </ span > < span class ="o " > . </ span > < span class =" n " > device </ span > < span class =" p "> (</ span > < span class =" s2 " > "cpu" </ span > < span class =" p " > )</ span >
616+ < span class ="k "> return</ span > < span class ="n "> _default_torch_device </ span > < span class ="p "> ()</ span >
606617
607618 < span class ="k "> def</ span > < span class ="w "> </ span > < span class ="nf "> _to_tensor</ span > < span class ="p "> (</ span > < span class ="bp "> self</ span > < span class ="p "> ,</ span > < span class ="n "> x</ span > < span class ="p "> :</ span > < span class ="n "> Any</ span > < span class ="p "> )</ span > < span class ="o "> -></ span > < span class ="n "> torch</ span > < span class ="o "> .</ span > < span class ="n "> Tensor</ span > < span class ="p "> :</ span >
608619 < span class ="n "> arr</ span > < span class ="o "> =</ span > < span class ="n "> _as_2d_numpy</ span > < span class ="p "> (</ span > < span class ="n "> x</ span > < span class ="p "> )</ span >
@@ -671,7 +682,7 @@ <h1>Source code for spflow.interfaces.sklearn</h1><div class="highlight"><pre>
671682
672683 </ aside >
673684 </ div >
674- </ div > < script src ="../../../_static/documentation_options.js?v=8d563738 "> </ script >
685+ </ div > < script src ="../../../_static/documentation_options.js?v=1ed6394b "> </ script >
675686 < script src ="../../../_static/doctools.js?v=fd6eb6e6 "> </ script >
676687 < script src ="../../../_static/sphinx_highlight.js?v=6ffebe34 "> </ script >
677688 < script src ="../../../_static/scripts/furo.js?v=46bd48cc "> </ script >
0 commit comments