Skip to content

Commit b344384

Browse files
committed
use camelcase for class name
1 parent 37e3ea8 commit b344384

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

pymc3/distributions/bart.py

+3-2
Original file line numberDiff line numberDiff line change
@@ -52,7 +52,7 @@ def __init__(self, X, Y, m=200, alpha=0.25, split_prior=None, *args, **kwargs):
5252
self.num_observations = X.shape[0]
5353
self.num_variates = X.shape[1]
5454
self.available_predictors = list(range(self.num_variates))
55-
self.ssv = sample_splitting_variable(split_prior, self.num_variates)
55+
self.ssv = SampleSplittingVariable(split_prior, self.num_variates)
5656
self.m = m
5757
self.alpha = alpha
5858
self.trees = self.init_list_of_trees()
@@ -174,6 +174,7 @@ def predict(self, X_new):
174174
trees = self.all_trees
175175
num_observations = X_new.shape[0]
176176
pred = np.zeros((len(trees), num_observations))
177+
np.random.randint(len(trees))
177178
for draw, trees_to_sum in enumerate(trees):
178179
new_Y = np.zeros(X_new.shape[0])
179180
for tree in trees_to_sum:
@@ -231,7 +232,7 @@ def discrete_uniform_sampler(upper_value):
231232
return int(np.random.random() * upper_value)
232233

233234

234-
class sample_splitting_variable:
235+
class SampleSplittingVariable:
235236
def __init__(self, prior, num_variates):
236237
self.prior = prior
237238
self.num_variates = num_variates

0 commit comments

Comments
 (0)