Skip to content

Commit 0cf7e1c

Browse files
committed
Revert "Pickle support"
This reverts commit d8cfb3c.
1 parent d8cfb3c commit 0cf7e1c

File tree

1 file changed

+0
-39
lines changed
  • pyTsetlinMachineParallel

1 file changed

+0
-39
lines changed

pyTsetlinMachineParallel/tm.py

Lines changed: 0 additions & 39 deletions
Original file line numberDiff line numberDiff line change
@@ -117,19 +117,6 @@ def __init__(self, number_of_clauses, T, s, patch_dim, boost_true_positive_feedb
117117
else:
118118
self.s_range = s
119119

120-
def __getstate__(self):
121-
state = self.__dict__.copy()
122-
state['mc_ctm_state'] = self.get_state()
123-
del state['mc_ctm']
124-
if 'encoded_X' in state:
125-
del state['encoded_X']
126-
return state
127-
128-
def __setstate__(self, state):
129-
self.__dict__.update(state)
130-
self.mc_ctm = _lib.CreateMultiClassTsetlinMachine(self.number_of_classes, self.number_of_clauses, self.number_of_features, self.number_of_patches, self.number_of_ta_chunks, self.number_of_state_bits, self.T, self.s, self.s_range, self.boost_true_positive_feedback, self.weighted_clauses, self.clause_drop_p, self.literal_drop_p)
131-
self.set_state(state['mc_ctm_state'])
132-
133120
def __del__(self):
134121
if self.mc_ctm != None:
135122
_lib.mc_tm_destroy(self.mc_ctm)
@@ -250,19 +237,6 @@ def __init__(self, number_of_clauses, T, s, boost_true_positive_feedback=1, numb
250237
else:
251238
self.s_range = s
252239

253-
def __getstate__(self):
254-
state = self.__dict__.copy()
255-
state['mc_tm_state'] = self.get_state()
256-
del state['mc_tm']
257-
if 'encoded_X' in state:
258-
del state['encoded_X']
259-
return state
260-
261-
def __setstate__(self, state):
262-
self.__dict__.update(state)
263-
self.mc_tm = _lib.CreateMultiClassTsetlinMachine(self.number_of_classes, self.number_of_clauses, self.number_of_features, 1, self.number_of_ta_chunks, self.number_of_state_bits, self.T, self.s, self.s_range, self.boost_true_positive_feedback, self.weighted_clauses, self.clause_drop_p, self.literal_drop_p)
264-
self.set_state(state['mc_tm_state'])
265-
266240
def __del__(self):
267241
if self.mc_tm != None:
268242
_lib.mc_tm_destroy(self.mc_tm)
@@ -374,19 +348,6 @@ def __init__(self, number_of_clauses, T, s, boost_true_positive_feedback=1, numb
374348
else:
375349
self.s_range = s
376350

377-
def __getstate__(self):
378-
state = self.__dict__.copy()
379-
state['rtm_state'] = self.get_state()
380-
del state['rtm']
381-
if 'encoded_X' in state:
382-
del state['encoded_X']
383-
return state
384-
385-
def __setstate__(self, state):
386-
self.__dict__.update(state)
387-
self.rtm = _lib.CreateTsetlinMachine(self.number_of_clauses, self.number_of_features, 1, self.number_of_ta_chunks, self.number_of_state_bits, self.T, self.s, self.s_range, self.boost_true_positive_feedback, self.weighted_clauses)
388-
self.set_state(state['rtm_state'])
389-
390351
def __del__(self):
391352
if self.rtm != None:
392353
_lib.tm_destroy(self.rtm)

0 commit comments

Comments
 (0)