Public Member Functions | |
def | __init__ |
def | analyze_testing_results |
def | build |
def | cross_valid |
def | load |
def | plot_testing_results |
Takes the trajectory list from eval_predicts and plots it. | |
def | predict |
def | train |
Public Attributes | |
fos | |
Private Member Functions | |
def | _fold_predictions |
Definition at line 19 of file classifiers.py.
Definition at line 21 of file classifiers.py.
def pr2_overhead_grasping.classifiers.AbstractClassifier._fold_predictions | ( | self, | |
compiled_dataset, | |||
num_folds, | |||
seed | |||
) | [private] |
Definition at line 57 of file classifiers.py.
def pr2_overhead_grasping.classifiers.AbstractClassifier.analyze_testing_results | ( | self, | |
traj_list | |||
) |
Definition at line 126 of file classifiers.py.
def pr2_overhead_grasping.classifiers.AbstractClassifier.build | ( | self, | |
compiled_dataset, | |||
filename | |||
) |
Reimplemented in pr2_overhead_grasping.classifiers.RFBreimanClassifier.
Definition at line 35 of file classifiers.py.
def pr2_overhead_grasping.classifiers.AbstractClassifier.cross_valid | ( | self, | |
compilied_dataset, | |||
num_folds = 10 , |
|||
seed = 1 |
|||
) |
Definition at line 41 of file classifiers.py.
def pr2_overhead_grasping.classifiers.AbstractClassifier.load | ( | self, | |
filename | |||
) |
Reimplemented in pr2_overhead_grasping.classifiers.RFBreimanClassifier.
Definition at line 38 of file classifiers.py.
def pr2_overhead_grasping.classifiers.AbstractClassifier.plot_testing_results | ( | self, | |
traj_list | |||
) |
Takes the trajectory list from eval_predicts and plots it.
each trajectory is stacked in a horizontal lines. Vertical placement indicates predicted result. Color indicates actual result. Collision trajectories are offset so that the colllision occurs at t = 0 Empty grasps are offset so that the end of the trajectory is at t = 0 Green = No collision, Red = External collision, Blue = Table collision
Definition at line 173 of file classifiers.py.
def pr2_overhead_grasping.classifiers.AbstractClassifier.predict | ( | self, | |
instance | |||
) |
Reimplemented in pr2_overhead_grasping.classifiers.RFBreimanClassifier.
Definition at line 31 of file classifiers.py.
def pr2_overhead_grasping.classifiers.AbstractClassifier.train | ( | self, | |
compiled_dataset | |||
) |
Reimplemented in pr2_overhead_grasping.classifiers.RFBreimanRefinedClassifier, and pr2_overhead_grasping.classifiers.RFBreimanClassifier.
Definition at line 28 of file classifiers.py.
Definition at line 21 of file classifiers.py.